Upload 5 files

app.py

@@ -0,0 +1,86 @@
+import streamlit as st
+import pandas as pd
+from sklearn.feature_extraction.text import TfidfVectorizer
+from sklearn.metrics.pairwise import cosine_similarity
+import numpy as np
+
+# Load Q&A data
+def load_data():
+    return pd.read_csv('qna.csv', encoding='latin1', delimiter=';')
+
+def get_most_similar_question(new_sentence, vectorizer, tfidf_matrix, questions, answers):
+    new_tfidf = vectorizer.transform([new_sentence])
+
+    similarities = cosine_similarity(new_tfidf, tfidf_matrix)
+
+    most_similar_index = np.argmax(similarities)
+
+    similarity_percentage = similarities[0, most_similar_index] * 100
+
+    return answers[most_similar_index], similarity_percentage
+
+def answer_the_question(new_sentence, vectorizer, tfidf_matrix, questions, answers):
+    most_similar_answer, similarity_percentage = get_most_similar_question(new_sentence, vectorizer, tfidf_matrix, questions, answers)
+    if similarity_percentage > 70:
+        return most_similar_answer
+    else:
+        return 'Sorry, I am not aware of this information :('
+
+def main():
+    
+    st.markdown(
+        "<h1 style='text-align: center; color: #269A96;'>Unicorn</h1>",
+        unsafe_allow_html=True
+    )
+
+    # Load Q&A data
+    data = load_data()
+    questions = data['question'].tolist()
+    answers = data['answer'].tolist()
+
+    # Create TF-IDF vectorizer
+    vectorizer = TfidfVectorizer()
+    tfidf_matrix = vectorizer.fit_transform(questions)
+
+    # Get user input
+    user_question = st.text_input("Ask me a question:", key="user_question")
+
+    # Apply CSS to style the text input field
+    st.markdown(
+        """
+        <style>
+        .st-cc {
+            background-color: #269A96 !important;
+            color: #FFFFFF !important;
+        }
+        .st-bk {
+            color: #FFFFFF !important;
+        }
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+
+    # Process the question when submitted
+    if st.button("Submit", key="submit_button", help="Submit your question"):
+        if user_question:
+            response = answer_the_question(user_question, vectorizer, tfidf_matrix, questions, answers)
+            st.write(response)
+        else:
+            st.warning("Please ask a question.")
+
+    # Apply CSS to style the button
+    st.markdown(
+        """
+        <style>
+        .css-1bkz6v6 {
+            background-color: #a5eae9;
+            color: #FFFFFF;
+        }
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+
+if __name__ == "__main__":
+    main()

qna.csv

@@ -0,0 +1,401 @@
+question;answer;;
+who is pm of India;Modi;;
+Who is Indian prime minister;Modi;;
+who is the leader of BJP;Modi;;
+Who is Indian pm;Modi;;
+who created you;I was made by Mr. Ninad Mandavkar. He is a Masters Graduate from TU Ilmenau. He is ML enthusiast and his current interests lies in NLP, Deep Learning & Computer Vision;;
+who developed you;I was made by Mr. Ninad Mandavkar. He is a Masters Graduate from TU Ilmenau. He is ML enthusiast and his current interests lies in NLP, Deep Learning & Computer Vision;;
+who discovred you;I was made by Mr. Ninad Mandavkar. He is a Masters Graduate from TU Ilmenau. He is ML enthusiast and his current interests lies in NLP, Deep Learning & Computer Vision;;
+who made you;I was made by Mr. Ninad Mandavkar. He is a Masters Graduate from TU Ilmenau. He is ML enthusiast and his current interests lies in NLP, Deep Learning & Computer Vision;;
+Who is harry potter?;Harry Potter is a fictional character written by J.K. Rowling  and he is the lead hero of a 7 parter series by the same name 'Harry Potter'.;;
+Who is JK rowling;J. K. Rowling is  a British author who created the series 'Harry Potter'. Rowling was born 31 July 1965 at Chipping Sodbury Cottage Hospital in Chipping Sodbury, Yate, Gloucesterhire England. She grew up in Tutshill, Gloucestershire, and went to school at Wyedean Comprehensive. She earned a degree in French and Classics at the University of Exeter. ;;
+Capital of Afghanistan;Kabul;;
+Capital of Albania;Tirana;;
+Capital of Algeria;Algiers;;
+Capital of Andorra;Andorra la Vella;;
+Capital of Angola;Luanda;;
+Capital of Antigua and Barbuda;St. John's;;
+Capital of Argentina;Buenos Aires;;
+Capital of Armenia;<;;
+Capital of Australia;Canberra;;
+Capital of Austria;Vienna;;
+Capital of Azerbaijan;Baku;;
+Capital of The Bahamas;Nassau;;
+Capital of Bahrain;Manama;;
+Capital of Bangladesh;Dhaka;;
+Capital of Barbados;Bridgetown;;
+Capital of Belarus;Minsk;;
+Capital of Belgium;Brussels;;
+Capital of Belize;Belmopan;;
+Capital of Benin;Porto-Novo;;
+Capital of Bhutan;Thimphu;;
+Capital of Bolivia;Sucre;;
+Capital of Bosnia and Herzegovina;Sarajevo;;
+Capital of Botswana;Gaborone;;
+Capital of Brazil;Bras�lia;;
+Capital of Brunei;Bandar Seri Begawan;;
+Capital of Bulgaria;Sofia;;
+Capital of Burkina Faso;Ouagadougou;;
+Capital of Burundi;Gitega;;
+Capital of Cabo Verde;Praia;;
+Capital of Cambodia;Phnom Penh;;
+Capital of Cameroon;Yaound�;;
+Capital of Canada;Ottawa;;
+Capital of Central African Republic;Bangui;;
+Capital of Chad;N'Djamena;;
+Capital of Chile;Santiago;;
+Capital of China;Beijing;;
+Capital of Colombia;Bogot�;;
+Capital of Comoros;Moroni;;
+Capital of Congo;Brazzaville;;
+Capital of Costa Rica;San Jos�;;
+Capital of Croatia;Zagreb;;
+Capital of Cuba;Havana;;
+Capital of Cyprus;Nicosia;;
+Capital of Czech Republic;Prague;;
+Capital of Democratic Republic of the Congo;Kinshasa;;
+Capital of Denmark;Copenhagen;;
+Capital of Djibouti;Djibouti (city);;
+Capital of Dominica;Roseau;;
+Capital of Dominican Republic;Santo Domingo;;
+Capital of East Timor;Dili;;
+Capital of Ecuador;Quito;;
+Capital of Egypt;Cairo;;
+Capital of El Salvador;San Salvador;;
+Capital of Equatorial Guinea;Malabo;;
+Capital of Eritrea;Asmara;;
+Capital of Estonia;Tallinn;;
+Capital of Eswatini;Mbabane (executive); Lobamba (legislative);
+Capital of Ethiopia;Addis Ababa;;
+Capital of Fiji;Suva;;
+Capital of Finland;Helsinki;;
+Capital of France;Paris;;
+Capital of Gabon;Libreville;;
+Capital of The Gambia;Banjul;;
+Capital of Georgia;Tbilisi;;
+Capital of Germany;Berlin;;
+Capital of Ghana;Accra;;
+Capital of Greece;Athens;;
+Capital of Grenada;St. George's;;
+Capital of Guatemala;Guatemala City;;
+Capital of Guinea;Conakry;;
+Capital of Guinea-Bissau;Bissau;;
+Capital of Guyana;Georgetown;;
+Capital of Haiti;Port-au-Prince;;
+Capital of Honduras;Tegucigalpa;;
+Capital of Hungary;Budapest;;
+Capital of Iceland;Reykjav�k;;
+Capital of India;New Delhi;;
+Capital of Indonesia;Jakarta;;
+Capital of Iran;Tehran;;
+Capital of Iraq;Baghdad;;
+Capital of Ireland;Dublin;;
+Capital of Israel;Jerusalem;;
+Capital of Italy;Rome;;
+Capital of Ivory Coast;Yamoussoukro (political capital); Abidjan (economic capital);
+Capital of Jamaica;Kingston;;
+Capital of Japan;Tokyo;;
+Capital of Jordan;Amman;;
+Capital of Kazakhstan;Nur-Sultan (formerly Astana);;
+Capital of Kenya;Nairobi;;
+Capital of Kiribati;South Tarawa;;
+Capital of Kosovo;Pristina;;
+Capital of Kuwait;Kuwait City;;
+Capital of Kyrgyzstan;Bishkek;;
+Capital of Laos;Vientiane;;
+Capital of Latvia;Riga;;
+Capital of Lebanon;Beirut;;
+Capital of Lesotho;Maseru;;
+Capital of Liberia;Monrovia;;
+Capital of Libya;Tripoli;;
+Capital of Liechtenstein;Vaduz;;
+Capital of Lithuania;Vilnius;;
+Capital of Luxembourg;Luxembourg City;;
+Capital of Madagascar;Antananarivo;;
+Capital of Malawi;Lilongwe;;
+Capital of Malaysia;Kuala Lumpur;;
+Capital of Maldives;Mal�;;
+Capital of Mali;Bamako;;
+Capital of Malta;Valletta;;
+Capital of Marshall Islands;Majuro;;
+Capital of Mauritania;Nouakchott;;
+Capital of Mauritius;Port Louis;;
+Capital of Mexico;Mexico City;;
+Capital of Federated States of Micronesia;Palikir;;
+Capital of Moldova;Chi?in?u;;
+Capital of Monaco;Monaco;;
+Capital of Mongolia;Ulaanbaatar;;
+Capital of Montenegro;Podgorica;;
+Capital of Morocco;Rabat;;
+Capital of Mozambique;Maputo;;
+Capital of Myanmar;Naypyidaw;;
+Capital of Namibia;Windhoek;;
+Capital of Nauru;Yaren (de facto);;
+Capital of Nepal;Kathmandu;;
+Capital of Netherlands;Amsterdam (official); The Hague (seat of government);
+Capital of New Zealand;Wellington;;
+Capital of Nicaragua;Managua;;
+Capital of Niger;Niamey;;
+Capital of Nigeria;Abuja;;
+Capital of North Korea;Pyongyang;;
+Capital of North Macedonia;Skopje;;
+Capital of Norway;Oslo;;
+Capital of Oman;Muscat;;
+Capital of Pakistan;Islamabad;;
+Capital of Palau;Ngerulmud;;
+Capital of Panama;Panama City;;
+Capital of Papua New Guinea;Port Moresby;;
+Capital of Paraguay;Asunci�n;;
+Capital of Peru;Lima;;
+Capital of Philippines;Manila;;
+Capital of Poland;Warsaw;;
+Capital of Portugal;Lisbon;;
+Capital of Qatar;Doha;;
+Capital of Romania;Bucharest;;
+Capital of Russia;Moscow;;
+Capital of Rwanda;Kigali;;
+Capital of Saint Kitts and Nevis;Basseterre;;
+Capital of Saint Lucia;Castries;;
+Capital of Saint Vincent and the Grenadines;Kingstown;;
+Capital of Samoa;Apia;;
+Capital of San Marino;San Marino;;
+Capital of Sao Tome and Principe;S�o Tom�;;
+Capital of Saudi Arabia;Riyadh;;
+Capital of Senegal;Dakar;;
+Capital of Serbia;Belgrade;;
+Capital of Seychelles;Victoria;;
+Capital of Sierra Leone;Freetown;;
+Capital of Singapore;Singapore;;
+Capital of Slovakia;Bratislava;;
+Capital of Slovenia;Ljubljana;;
+Capital of Solomon Islands;Honiara;;
+Capital of Somalia;Mogadishu;;
+Capital of South Africa;Pretoria (executive); Bloemfontein (judicial); Cape Town (legislative)
+Capital of South Korea;Seoul;;
+Capital of South Sudan;Juba;;
+Capital of Spain;Madrid;;
+Capital of Sri Lanka;Sri Jayawardenepura Kotte (official); Colombo (commercial);
+Capital of Sudan;Khartoum;;
+Capital of Sweden;Stockholm;;
+Capital of Switzerland;Bern;;
+Capital of Syria;Damascus;;
+Capital of Taiwan;Taipei;;
+Capital of Tajikistan;Dushanbe;;
+Capital of Tanzania;Dodoma (official); Dar es Salaam (administrative);
+Capital of Thailand;Bangkok;;
+Capital of Togo;Lom�;;
+Capital of Tonga;Nuku?alofa;;
+Capital of Trinidad and Tobago;Port of Spain;;
+Capital of Tunisia;Tunis;;
+Capital of Turkey;Ankara;;
+Capital of Turkmenistan;Ashgabat;;
+Capital of Tuvalu;Funafuti;;
+Capital of Uganda;Kampala;;
+Capital of Ukraine;Kyiv;;
+Capital of United Arab Emirates;Abu Dhabi;;
+Capital of United Kingdom;London;;
+Capital of United States;Washington; D.C.;
+Capital of Uruguay;Montevideo;;
+Capital of Uzbekistan;Tashkent;;
+Capital of Vanuatu;Port Vila;;
+Capital of Vatican City;Vatican City;;
+Capital of Venezuela;Caracas;;
+Capital of Vietnam;Hanoi;;
+Capital of Yemen;Sana'a (de jure); Aden (temporary); Sana'a (Houthi-led Yemen)
+Capital of Zambia;Lusaka;;
+Capital of Zimbabwe;Harare;;
+President of Afghanistan;Ashraf Ghani;;
+President of Albania;Ilir Meta;;
+President of Algeria;Abdelmadjid Tebboune;;
+President of Andorra;Xavier Espot Zamora;;
+President of Angola;Jo�o Louren�o;;
+President of Antigua and Barbuda;Rodney Williams;;
+President of Argentina;Alberto Fern�ndez;;
+President of Armenia;Armen Sarkissian;;
+President of Australia;Scott Morrison;;
+President of Austria;Alexander Van der Bellen;;
+President of Azerbaijan;Ilham Aliyev;;
+President of The Bahamas;Hubert Minnis;;
+President of Bahrain;Hamad bin Isa Al Khalifa;;
+President of Bangladesh;Abdul Hamid;;
+President of Barbados;Sandra Mason;;
+President of Belarus;Alexander Lukashenko;;
+President of Belgium;King Philippe;;
+President of Belize;Johnny Brice�o;;
+President of Benin;Patrice Talon;;
+President of Bhutan;Jigme Khesar Namgyel Wangchuck;;
+President of Bolivia;Luis Arce;;
+President of Bosnia and Herzegovina;�efik D�aferovi? (Chairman of the Presidency);;
+President of Botswana;Mokgweetsi Masisi;;
+President of Brazil;Jair Bolsonaro;;
+President of Brunei;Hassanal Bolkiah;;
+President of Bulgaria;Rumen Radev;;
+President of Burkina Faso;Roch Marc Christian Kabor�;;
+President of Burundi;�variste Ndayishimiye;;
+President of Cabo Verde;Jorge Carlos Fonseca;;
+President of Cambodia;King Norodom Sihamoni;;
+President of Cameroon;Paul Biya;;
+President of Canada;Justin Trudeau;;
+President of Central African Republic;Faustin-Archange Touad�ra;;
+President of Chad;Idriss D�by;;
+President of Chile;Sebasti�n Pi�era;;
+President of China;Xi Jinping;;
+President of Colombia;Iv�n Duque M�rquez;;
+President of Comoros;Azali Assoumani;;
+President of Congo;Denis Sassou Nguesso;;
+President of Costa Rica;Carlos Alvarado Quesada;;
+President of Croatia;Zoran Milanovi?;;
+President of Cuba;Miguel D�az-Canel;;
+President of Cyprus;Nicos Anastasiades;;
+President of Czech Republic;Milo� Zeman;;
+President of Democratic Republic of the Congo;F�lix Tshisekedi;;
+President of Denmark;Queen Margrethe II;;
+President of Djibouti;Ismail Omar Guelleh;;
+President of Dominica;Charles Savarin;;
+President of Dominican Republic;Luis Abinader;;
+President of East Timor;Francisco Guterres;;
+President of Ecuador;Len�n Moreno;;
+President of Egypt;Abdel Fattah el-Sisi;;
+President of El Salvador;Nayib Bukele;;
+President of Equatorial Guinea;Teodoro Obiang Nguema Mbasogo;;
+President of Eritrea;Isaias Afwerki;;
+President of Estonia;Kersti Kaljulaid;;
+President of Eswatini;King Mswati III;;
+President of Ethiopia;Sahle-Work Zewde;;
+President of Fiji;Jioji Konrote;;
+President of Finland;Sauli Niinist�;;
+President of France;Emmanuel Macron;;
+President of Gabon;Ali Bongo Ondimba;;
+President of The Gambia;Adama Barrow;;
+President of Georgia;Salome Zourabichvili;;
+President of Germany;Frank-Walter Steinmeier;;
+President of Ghana;Nana Akufo-Addo;;
+President of Greece;Katerina Sakellaropoulou;;
+President of Grenada;Angus Mitchell;;
+President of Guatemala;Alejandro Giammattei;;
+President of Guinea;Alpha Cond�;;
+President of Guinea-Bissau;Umaro Sissoco Embal�;;
+President of Guyana;Irfaan Ali;;
+President of Haiti;Jovenel Mo�se;;
+President of Honduras;Juan Orlando Hern�ndez;;
+President of Hungary;J�nos �der;;
+President of Iceland;Gu�ni Th. J�hannesson;;
+President of India;Droupadi Murmu;;
+President of Indonesia;Joko Widodo;;
+President of Iran;Ebrahim Raisi;;
+President of Iraq;Barham Salih;;
+President of Ireland;Michael D. Higgins;;
+President of Israel;Reuven Rivlin;;
+President of Italy;Sergio Mattarella;;
+President of Ivory Coast;Alassane Ouattara;;
+President of Jamaica;Andrew Holness;;
+President of Japan;Naruhito;;
+President of Jordan;Abdullah II;;
+President of Kazakhstan;Kassym-Jomart Tokayev;;
+President of Kenya;Uhuru Kenyatta;;
+President of Kiribati;Taneti Mamau;;
+President of Kosovo;Vjosa Osmani;;
+President of Kuwait;Sabah  Al-Ahmad Al-Jaber Al-Sabah;;
+President of Kyrgyzstan;Sadyr Japarov;;
+President of Laos;Bounnhang Vorachith;;
+President of Latvia;Egils Levits;;
+President of Lebanon;Michel Aoun;;
+President of Lesotho;Letsie III;;
+President of Liberia;George Weah;;
+President of Libya;Mohamed al-Menfi;;
+President of Liechtenstein;Adrian Hasler;;
+President of Lithuania;Gitanas Nauseda;;
+President of Luxembourg;Henri;;
+President of Madagascar;Andry Rajoelina;;
+President of Malawi;Lazarus Chakwera;;
+President of Malaysia;Sultan Abdullah of Pahang;;
+President of Maldives;Ibrahim Mohamed Solih;;
+President of Mali;Assimi Go�ta;;
+President of Malta;George Vella;;
+President of Marshall Islands;David Kabua;;
+President of Mauritania;Mohamed Ould Ghazouani;;
+President of Mauritius;Prithvirajsing Roopun;;
+President of Mexico;Andr�s Manuel L�pez Obrador;;
+President of Federated States of Micronesia;David W. Panuelo;;
+President of Moldova;Maia Sandu;;
+President of Monaco;Prince Albert II;;
+President of Mongolia;Khaltmaagiin Battulga;;
+President of Montenegro;Milo ?ukanovi?;;
+President of Morocco;Mohammed VI;;
+President of Mozambique;Filipe Nyusi;;
+President of Myanmar;Myint Swe;;
+President of Namibia;Hage Geingob;;
+President of Nauru;Lionel Aingimea;;
+President of Nepal;Bidhya Devi Bhandari;;
+President of Netherlands;King Willem-Alexander;;
+President of New Zealand;Jacinda Ardern;;
+President of Nicaragua;Daniel Ortega;;
+President of Niger;Mohamed Bazoum;;
+President of Nigeria;Muhammadu Buhari;;
+President of North Korea;Kim Jong-un;;
+President of North Macedonia;Stevo Pendarovski;;
+President of Norway;Harald V;;
+President of Oman;Haitham bin Tariq Al Said;;
+President of Pakistan;Arif Alvi;;
+President of Palau;Surangel Whipps Jr.;;
+President of Panama;Laurentino Cortizo;;
+President of Papua New Guinea;James Marape;;
+President of Paraguay;Mario Abdo Ben�tez;;
+President of Peru;Pedro Castillo;;
+President of Philippines;Rodrigo Duterte;;
+President of Poland;Andrzej Duda;;
+President of Portugal;Marcelo Rebelo de Sousa;;
+President of Qatar;Tamim bin Hamad Al Thani;;
+President of Romania;Klaus Iohannis;;
+President of Russia;Vladimir Putin;;
+President of Rwanda;Paul Kagame;;
+President of Saint Kitts and Nevis;Timothy Harris;;
+President of Saint Lucia;Allen Chastanet;;
+President of Saint Vincent and the Grenadines;Ralph Gonsalves;;
+President of Samoa;Tuimalealiifano Vaaleto'a Sualauvi II;;
+President of San Marino;Gian Carlo Venturini;;
+President of Sao Tome and Principe;Evaristo Carvalho;;
+President of Saudi Arabia;Salman bin Abdulaziz Al Saud;;
+President of Senegal;Macky Sall;;
+President of Serbia;Aleksandar Vu?i?;;
+President of Seychelles;Wavel Ramkalawan;;
+President of Sierra Leone;Julius Maada Bio;;
+President of Singapore;Halimah Yacob;;
+President of Slovakia;Zuzana ?aputov�;;
+President of Slovenia;Borut Pahor;;
+President of Solomon Islands;David Kabua;;
+President of Somalia;Mohamed Abdullahi Mohamed;;
+President of South Africa;Cyril Ramaphosa;;
+President of South Korea;Moon Jae-in;;
+President of South Sudan;Salva Kiir Mayardit;;
+President of Spain;Pedro S�nchez;;
+President of Sri Lanka;Gotabaya Rajapaksa;;
+President of Sudan;Abdel Fattah al-Burhan;;
+President of Sweden;Carl XVI Gustaf;;
+President of Switzerland;Guy Parmelin;;
+President of Syria;Bashar al-Assad;;
+President of Taiwan;Tsai Ing-wen;;
+President of Tajikistan;Emomali Rahmon;;
+President of Tanzania;Samia Suluhu;;
+President of Thailand;Maha Vajiralongkorn;;
+President of Togo;Faure Gnassingb�;;
+President of Tonga;King Tupou VI;;
+President of Trinidad and Tobago;Paula-Mae Weekes;;
+President of Tunisia;Kais Saied;;
+President of Turkey;Recep Tayyip Erdo?an;;
+President of Turkmenistan;Gurbanguly Berdimuhamedow;;
+President of Tuvalu;Taneti Mamau;;
+President of Uganda;Yoweri Museveni;;
+President of Ukraine;Volodymyr Zelensky;;
+President of  United Arab Emirates;Khalifa bin Zayed Al Nahyan;;
+President of United Kingdom;Queen Elizabeth II;;
+President of United States;Joe Biden;;
+President of Uruguay;Luis Lacalle Pou;;
+President of Uzbekistan;Shavkat Mirziyoyev;;
+President of Vanuatu;Tallis Obed Moses;;
+President of Vatican City;Pope Francis;;
+President of Venezuela;Nicol�s Maduro;;
+President of Vietnam;Nguy?n Ph� Tr?ng;;
+President of Yemen;Abdrabbuh Mansur Hadi;;
+President of Zambia;Edgar Lungu;;
+President of Zimbabwe;Emmerson Mnangagwa;;

train.py

@@ -0,0 +1,55 @@
+import streamlit as st
+import pandas as pd
+from sklearn.feature_extraction.text import TfidfVectorizer
+from sklearn.metrics.pairwise import cosine_similarity
+import numpy as np
+
+# Load Q&A data
+@st.cache
+def load_data():
+    return pd.read_csv('qna.csv', encoding='utf-8', delimiter=';')
+
+def get_most_similar_question(new_sentence, vectorizer, tfidf_matrix, questions, answers):
+    new_tfidf = vectorizer.transform([new_sentence])
+
+    similarities = cosine_similarity(new_tfidf, tfidf_matrix)
+
+    most_similar_index = np.argmax(similarities)
+
+    similarity_percentage = similarities[0, most_similar_index] * 100
+
+    return answers[most_similar_index], similarity_percentage
+
+def answer_the_question(new_sentence, vectorizer, tfidf_matrix, questions, answers):
+    most_similar_answer, similarity_percentage = get_most_similar_question(new_sentence, vectorizer, tfidf_matrix, questions, answers)
+    if similarity_percentage > 70:
+        return most_similar_answer
+    else:
+        return 'Sorry, I am not aware of this information :('
+
+def main():
+    st.title("Q&A Chatbot")
+
+    # Load Q&A data
+    data = load_data()
+    questions = data['question'].tolist()
+    answers = data['answer'].tolist()
+
+    # Create TF-IDF vectorizer
+    vectorizer = TfidfVectorizer()
+    tfidf_matrix = vectorizer.fit_transform(questions)
+
+    # Get user input
+    user_question = st.text_input("Ask me a question:")
+
+    # Process the question when submitted
+    if st.button("Submit"):
+        if user_question:
+            response = answer_the_question(user_question, vectorizer, tfidf_matrix, questions, answers)
+            st.write("Answer:")
+            st.write(response)
+        else:
+            st.warning("Please ask a question.")
+
+if __name__ == "__main__":
+    main()