Update app.py

app.py

@@ -8,7 +8,7 @@ import time
 load_dotenv()
 
 # Retrieve API key from environment variable
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")  # Securely retrieve the API key from the environment
+GEMINI_API_KEY = "AIzaSyA0SnGcdEuesDusLiM93N68-vaFF14RCYg" # public api 
 
 # Configure Google Gemini API
 genai.configure(api_key=GEMINI_API_KEY)
@@ -44,7 +44,7 @@ def generate_response(user_input, chat_history):
     )
 
     # Add user input to history
-    chat_history.append(("User", user_input))
+    chat_history.append(user_input)
 
     # Limit history length to the last 10 messages
     chat_history = chat_history[-10:]
@@ -56,29 +56,21 @@ def generate_response(user_input, chat_history):
             chat_session = model.start_chat()
 
             # Send the entire chat history as the first message
-            messages = [f"{sender}: {message}" for sender, message in chat_history]
-            response = chat_session.send_message("\n".join(messages))
-            
-            # Add the response to the chat history
-            chat_history.append(("Shadow", response.text))
-
-            return chat_history, chat_history
+            response = chat_session.send_message("\n".join(chat_history))
+            return response.text, chat_history
 
         except Exception as e:
             if attempt < retry_attempts - 1:
                 time.sleep(2)  # Delay before retrying
                 continue
             else:
-                error_message = f"Error after {retry_attempts} attempts: {str(e)}"
-                chat_history.append(("Shadow", error_message))
-                return chat_history, chat_history
+                return f"Error after {retry_attempts} attempts: {str(e)}", chat_history
 
 # Build the Gradio interface
 with gr.Blocks(theme="Hev832/Applio") as iface:
-    gr.Markdown("<center><h1>Chat with Shadow The Hedgehog</h1></center>")
-    chat_input = gr.Textbox(lines=2, label="Talk to Shadow", placeholder="Enter your message here...")
+    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.Chatbot(label="Response")
+    response_output = gr.Textbox(label="Response")
 
     # Define the layout and components
     generate_button = gr.Button("Generate Response")