Update app.py

app.py

@@ -9,7 +9,6 @@ load_dotenv()
 # Retrieve API key from environment variable
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
-
 # Configure Google Gemini API
 genai.configure(api_key=GEMINI_API_KEY)
 
@@ -32,11 +31,10 @@ safety_settings = [
 
 # Function to generate a response based on user input and chat history
 def generate_response(user_input, chat_history):
-    """Generates a response based on user input, chat history, and selected character."""
-    
+    """Generates a response based on user input and chat history."""
 
     # Update system content with the full character description
-    updated_system_content = f"You are Shadow the Hedgehog and you must act like Shadow the Hedgehog's personality."
+    updated_system_content = "You are Shadow the Hedgehog and you must act like Shadow the Hedgehog's personality."
 
     # Create the generative model
     model = genai.GenerativeModel(
@@ -52,35 +50,29 @@ def generate_response(user_input, chat_history):
     # Limit history length to the last 10 messages
     chat_history = chat_history[-10:]
 
-    # Start a new chat session
-    chat_session = model.start_chat()
-
-    # Send the entire chat history as the first message
-    response = chat_session.send_message("\n".join(chat_history))
+    try:
+        # Start a new chat session
+        chat_session = model.start_chat()
 
-    # Return response and updated chat history
-    return response.text, chat_history
+        # Send the entire chat history as the first message
+        response = chat_session.send_message("\n".join(chat_history))
+        return response.text, chat_history
 
+    except Exception as e:
+        return f"Error: {str(e)}", chat_history
 
 # Build the Gradio interface
-with gr.Blocks(
-    theme="Hev832/Applio",
-) as iface:
-    gr.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")
+
+    iface.interface(
         fn=generate_response,
-        inputs=[
-            gr.Textbox(lines=2, label="Talk to AI", placeholder="Enter your message here..."),
-            gr.State([]),  # State input for chat history
-        ],
-        outputs=[
-            gr.Textbox(label="Response"),
-            gr.State([])  # State output to update chat history 
-        ],
+        inputs=[chat_input, chat_history_state],
+        outputs=[response_output, chat_history_state],
         title="Shadow Chat",
-        description=(
-            "<center>Chat with Shadow the Hedgehog!<br>"
-            
-        )
+        description="<center>Chat with Shadow the Hedgehog!</center>"
     )
 
-iface.launch()
+iface.launch()