Update app.py

app.py

@@ -1,44 +1,36 @@
-#Hello! It seems like you want to import the Streamlit library in Python. Streamlit is a powerful open-source framework used for building web applications with interactive data visualizations and machine learning models. To import Streamlit, you'll need to ensure that you have it installed in your Python environment.
-#Once you have Streamlit installed, you can import it into your Python script using the import statement,
-
 import streamlit as st
-
-#from langchain_openai import OpenAI
 from langchain.llms import HuggingFaceEndpoint
 
-#When deployed on huggingface spaces, this values has to be passed using Variables & Secrets setting, as shown in the video :)
-#import os
-#os.environ["OPENAI_API_KEY"] = "sk-PLfFwPq6y24234234234FJ1Uc234234L8hVowXdt"
-
-#Function to return the response
+# Function to return the response
 def load_answer(question):
     llm = HuggingFaceEndpoint(
-    repo_id="mistralai/Mistral-7B-Instruct-v0.2") # Model link : https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2
+        repo_id="mistralai/Mistral-7B-Instruct-v0.2")  # Model link: https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2
 
-    answer=llm.invoke(question)
+    if question.strip() == "":
+        return "Question cannot be empty. Please enter a valid question."
+    
+    answer = llm.invoke(question)
     return answer
 
-
-#App UI starts here
+# App UI starts here
 st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
 st.header("LangChain Demo by Sankar")
 
-#Gets the user input
+# Gets the user input
 def get_text():
     input_text = st.text_input("You: ", key="input")
     return input_text
 
+user_input = get_text()
 
-user_input=get_text()
-response = load_answer(user_input)
-
-submit = st.button('Generate')  
+submit = st.button('Generate')
 
-# If generate button is clicked and user_input is not empty
-if submit and user_input:
-    response = load_answer(user_input)
-    st.subheader("Answer:")
-    st.write(response)
-elif submit:
-    st.subheader("Answer:")
-    st.write("Please enter a question.")
+# If generate button is clicked
+if submit:
+    if user_input.strip() == "":
+        st.subheader("Answer:")
+        st.write("Question cannot be empty. Please enter a valid question.")
+    else:
+        response = load_answer(user_input)
+        st.subheader("Answer:")
+        st.write(response)