Update app.py

app.py

@@ -12,6 +12,9 @@ import traceback
 # Initialize embedding model
 sbert_model = SentenceTransformer('all-MiniLM-L6-v2')
 
+# Get API key from environment
+GEMINI_API_KEY = os.getenv('GAPI')
+
 # Data storage
 chunks = []
 faiss_index = None
@@ -61,12 +64,12 @@ def create_faiss_index(embeddings):
     
     return index
 
-def process_pdf(pdf_file_obj, api_key):
+def process_pdf(pdf_file_obj):
     """Process PDF and create FAISS index."""
     global chunks, faiss_index
 
-    if not api_key:
-        return None, [["System", "⚠️ Please set your Gemini API key first."]]
+    if not GEMINI_API_KEY:
+        return None, [["System", "⚠️ GAPI environment variable not set. Please configure your Gemini API key."]]
     
     if pdf_file_obj is None:
         return None, [["System", "📄 Please upload a PDF file."]]
@@ -146,7 +149,7 @@ def retrieve_relevant_chunks(query, top_k=3):
         print(f"Error in FAISS search: {str(e)}")
         return []
 
-def chat_fn(message, history, api_key):
+def chat_fn(message, history):
     """Handle chat interaction."""
     if not message.strip():
         return history, ""
@@ -154,8 +157,8 @@ def chat_fn(message, history, api_key):
     # Add user message to history
     history = history + [[message, None]]
 
-    if not api_key:
-        history[-1][1] = "⚠️ Please set your Gemini API key first."
+    if not GEMINI_API_KEY:
+        history[-1][1] = "⚠️ GAPI environment variable not set. Please configure your Gemini API key."
         return history, ""
 
     if not chunks or faiss_index is None:
@@ -164,7 +167,7 @@ def chat_fn(message, history, api_key):
 
     try:
         # Configure Gemini
-        genai.configure(api_key=api_key)
+        genai.configure(api_key=GEMINI_API_KEY)
         
         # Get relevant context using FAISS
         context_chunks = retrieve_relevant_chunks(message, top_k=5)
@@ -207,25 +210,18 @@ css = """
 """
 
 with gr.Blocks(css=css, title="📚 Chat with Your PDF") as demo:
-    api_key_state = gr.State("")
     
-    gr.Markdown("""
+    gr.Markdown(f"""
     # 📚 Chat with Your PDF (FAISS-Powered)
     Upload a PDF document and chat with it naturally. Now with FAISS for faster vector search!
+    
+    {"✅ API Key: Configured from environment" if GEMINI_API_KEY else "❌ API Key: Not found in GAPI environment variable"}
     """)
     
-    with gr.Row():
-        with gr.Column(scale=2):
-            api_key_input = gr.Textbox(
-                label="🔑 Gemini API Key", 
-                type="password", 
-                placeholder="Enter your API key here..."
-            )
-        with gr.Column(scale=1):
-            pdf_input = gr.File(
-                label="📄 Upload PDF", 
-                file_types=['.pdf']
-            )
+    pdf_input = gr.File(
+        label="📄 Upload PDF", 
+        file_types=['.pdf']
+    )
     
     # Chat interface
     chatbot = gr.Chatbot(
@@ -247,30 +243,21 @@ with gr.Blocks(css=css, title="📚 Chat with Your PDF") as demo:
         clear_btn = gr.Button("Clear Chat 🗑️")
     
     # Event handlers
-    def update_api_key(key):
-        return key
-    
-    api_key_input.change(
-        fn=update_api_key,
-        inputs=api_key_input,
-        outputs=api_key_state
-    )
-    
     pdf_input.upload(
         fn=process_pdf,
-        inputs=[pdf_input, api_key_state],
+        inputs=[pdf_input],
         outputs=[msg_input, chatbot]
     )
     
     submit_btn.click(
         fn=chat_fn,
-        inputs=[msg_input, chatbot, api_key_state],
+        inputs=[msg_input, chatbot],
         outputs=[chatbot, msg_input]
     )
     
     msg_input.submit(
         fn=chat_fn,
-        inputs=[msg_input, chatbot, api_key_state],
+        inputs=[msg_input, chatbot],
         outputs=[chatbot, msg_input]
     )