Update app.py

app.py

@@ -1,14 +1,12 @@
-import os
 import gradio as gr
 import google.generativeai as genai
 from dotenv import load_dotenv
-import time
 
 # Load environment variables from .env file
 load_dotenv()
 
 # Retrieve API key from environment variable
-GEMINI_API_KEY = "AIzaSyA0SnGcdEuesDusLiM93N68-vaFF14RCYg" # public api 
+GEMINI_API_KEY = "AIzaSyA0SnGcdEuesDusLiM93N68-vaFF14RCYg"  # public api 
 
 # Configure Google Gemini API
 genai.configure(api_key=GEMINI_API_KEY)
@@ -61,23 +59,10 @@ def generate_response(user_input, chat_history):
 
         except Exception as e:
             if attempt < retry_attempts - 1:
-                time.sleep(2)  # Delay before retrying
                 continue
             else:
                 return f"Error after {retry_attempts} attempts: {str(e)}", chat_history
 
-# Build the Gradio interface
-with gr.Blocks(theme="Hev832/Applio") as iface:
-    chat_input = gr.Textbox(lines=2, label="Talk to AI", placeholder="Enter your message here...")
-    chat_history_state = gr.State([])  # State input for chat history
-    response_output = gr.Textbox(label="Response")
-
-    # Define the layout and components
-    generate_button = gr.Button("Generate Response")
-    generate_button.click(
-        fn=generate_response,
-        inputs=[chat_input, chat_history_state],
-        outputs=[response_output, chat_history_state]
-    )
-
-iface.launch()
+# Build the Gradio interface using ChatInterface
+with gr.ChatInterface(fn=generate_response) as iface:
+    iface.launch()