Update app.py

app.py

@@ -27,20 +27,19 @@ def to_markdown(text):
     return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))
 
 
-def chat(chat_history):
-    """Generates a response based on the chat history.
+def chat(user_message):
+    """Generates a response to the user's message.
 
     Args:
-        chat_history (list): A list containing user messages and AI responses.
+        user_message (str): The user's message.
 
     Returns:
-        str: The AI's response to the latest user message.
+        str: The AI-generated response (or a message indicating unavailability).
     """
 
     if not genai:
         return "AI responses are currently unavailable. Please install `google-generativeai` for this functionality."
 
-    user_message = chat_history[-1]  # Get the latest user message
     try:
         response = model.generate_content(user_message, stream=True)
         for chunk in response:
@@ -52,10 +51,10 @@ def chat(chat_history):
 
 interface = gr.Interface(
     fn=chat,
-    inputs="chat",
+    inputs="textbox",
     outputs="textbox",
     title="Gradio Chat App",
-    description="Chat with an AI assistant (requires `google-generativeai`)"
+    description="Chat with an AI assistant (requires `google-generativeai`)",
 )
 
 interface.launch()
@@ -68,11 +67,8 @@ if __name__ == "__main__":
     genai.configure(api_key='AIzaSyCMBk81YmILNTok8hd6tYtJaevp1qbl6I0')  # Replace with your actual API key
     model = genai.GenerativeModel('gemini-pro')
 
-    chat_history = []
     while True:
         user_message = input("You: ")
-        chat_history.append(user_message)
-        response = chat(chat_history)
+        response = chat(user_message)
         print(f"AI: {response}")
-        chat_history.append(response)
         print("-" * 80)