Update app.py

app.py

@@ -4,9 +4,6 @@ import json
 import time
 import gradio as gr
 import logging
-
-
-
 import google.generativeai as genai
 
 from langgraph.graph import START, MessagesState, StateGraph
@@ -24,19 +21,11 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)
 
-# Pull from Hugging Face Secrets
-GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
-
-if not GEMINI_API_KEY:
-    raise RuntimeError("GEMINI_API_KEY not found in Hugging Face Secrets")
-
-# Ensure it's accessible globally
+# === HARD-CODED API KEY (for testing only) ===
+GEMINI_API_KEY = AIzaSyCZtDtt7K6g3IgR0GsQ2pil4xsu7Jrt_lk
 os.environ["GOOGLE_API_KEY"] = GEMINI_API_KEY
-
-# Configure Google GenAI
 genai.configure(api_key=GEMINI_API_KEY)
 
-
 # === Chat Storage ===
 HISTORY_FILE = "chat_history.json"
 
@@ -52,7 +41,7 @@ def save_all_sessions(sessions):
 
 sessions = load_all_sessions()
 
-# === Gemini LLM Chatbot ===
+# === Gemini Chatbot ===
 class GeminiChatbot:
     def __init__(self):
         self.setup_model()
@@ -93,10 +82,7 @@ class GeminiChatbot:
         self.app = workflow.compile(checkpointer=self.memory)
 
     def get_response(self, user_message, history, thread_id):
-        from langchain_core.messages import HumanMessage, AIMessage
-
         try:
-            # Format chat history for LangChain
             langchain_history = []
             for user, bot in history:
                 langchain_history.append(HumanMessage(content=user))
@@ -106,7 +92,6 @@ class GeminiChatbot:
             full_history = langchain_history + [input_msg]
             config = {"configurable": {"thread_id": thread_id}}
 
-            # Get final response
             response = self.app.invoke({"messages": full_history}, config)
             full_text = response["messages"][-1].content
 
@@ -120,7 +105,6 @@ class GeminiChatbot:
             logger.error(f"Response error: {e}")
             yield f"⚠ Error: {type(e).__name__} — {str(e)}"
 
-
 chatbot = GeminiChatbot()
 
 # === Gradio UI ===
@@ -159,7 +143,6 @@ def launch_interface():
         def get_dropdown_choices():
             return [f"PREVIOUS: {k}" for k in sessions if sessions[k]] + [f"NEW: {current_thread_id.value}"]
 
-        # UI
         new_chat_btn = gr.Button("New Chat", variant="primary")
         session_selector = gr.Dropdown(
             label="Chats",
@@ -176,7 +159,6 @@ def launch_interface():
 
         clear = gr.Button("Clear Current Chat")
 
-        # === Event Functions ===
         def start_new_chat():
             new_id = str(uuid.uuid4())
             sessions[new_id] = []
@@ -208,7 +190,7 @@ def launch_interface():
             save_all_sessions(sessions)
             return []
 
-        # === Bind Events ===
+        # Events
         new_chat_btn.click(start_new_chat, outputs=[current_thread_id, chatbot_ui, session_selector, session_selector])
         session_selector.change(switch_chat, inputs=session_selector, outputs=[current_thread_id, chatbot_ui, session_selector])
         send.click(respond, [msg, chatbot_ui, current_thread_id], [chatbot_ui]).then(lambda: "", None, msg)