Update app.py

app.py

@@ -8,7 +8,7 @@ import time
 load_dotenv()
 
 # Retrieve API key from environment variable
-GEMINI_API_KEY = "AIzaSyDpoauyPoIGqQlybQH0KdZoxZgf70oUXgA" # public api 
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")  # Securely retrieve the API key from the environment
 
 # 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_input)
+    chat_history.append(("User", user_input))
 
     # Limit history length to the last 10 messages
     chat_history = chat_history[-10:]
@@ -56,20 +56,27 @@ def generate_response(user_input, chat_history):
             chat_session = model.start_chat()
 
             # Send the entire chat history as the first message
-            response = chat_session.send_message("\n".join(chat_history))
-            return response.text, chat_history
+            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
 
         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
+                error_message = f"Error after {retry_attempts} attempts: {str(e)}"
+                chat_history.append(("Shadow", error_message))
+                return chat_history, chat_history
 
 # Build the Gradio interface
 with gr.Blocks(theme="Hev832/Applio") as iface:
-    gr.Markdown("<center><h1>Chat with Shadow The Hedgehog")
-    chat_input = gr.Textbox(lines=2, label="Talk to shadow", placeholder="Enter your message here...")
+    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_history_state = gr.State([])  # State input for chat history
     response_output = gr.Chatbot(label="Response")