Update app.py

app.py

@@ -17,11 +17,27 @@ import streamlit as st
 import requests
 from langchain.tools.retriever import create_retriever_tool
 
+# ------------------------------
+# Configuration
+# ------------------------------
+# Get DeepSeek API key from Hugging Face Space secrets
+DEEPSEEK_API_KEY = os.environ.get("DEEPSEEK_API_KEY")
+
+if not DEEPSEEK_API_KEY:
+    st.error("""
+    **Missing API Configuration**  
+    Please configure your DeepSeek API key in Hugging Face Space secrets:
+    1. Go to your Space's Settings
+    2. Click on 'Repository secrets'
+    3. Add a secret named DEEPSEEK_API_KEY
+    """)
+    st.stop()
+
 # Create directory for Chroma persistence
 os.makedirs("chroma_db", exist_ok=True)
 
 # ------------------------------
-# ChromaDB Client Configuration (Updated)
+# ChromaDB Client Configuration
 # ------------------------------
 chroma_client = chromadb.PersistentClient(path="chroma_db")
 
@@ -116,7 +132,7 @@ Otherwise, just answer directly.
 
     headers = {
         "Accept": "application/json",
-        "Authorization": f"Bearer sk-1cddf19f9dc4466fa3ecea6fe10abec0",
+        "Authorization": f"Bearer {DEEPSEEK_API_KEY}",
         "Content-Type": "application/json"
     }
     
@@ -127,14 +143,16 @@ Otherwise, just answer directly.
         "max_tokens": 1024
     }
     
-    response = requests.post(
-        "https://api.deepseek.com/v1/chat/completions",
-        headers=headers,
-        json=data,
-        verify=False
-    )
-    
-    if response.status_code == 200:
+    try:
+        response = requests.post(
+            "https://api.deepseek.com/v1/chat/completions",
+            headers=headers,
+            json=data,
+            verify=False,
+            timeout=30
+        )
+        response.raise_for_status()
+        
         response_text = response.json()['choices'][0]['message']['content']
         print("Raw response:", response_text)
         
@@ -150,8 +168,12 @@ Otherwise, just answer directly.
 
         else:
             return {"messages": [AIMessage(content=response_text)]}
-    else:
-        raise Exception(f"API call failed: {response.text}")
+
+    except Exception as e:
+        error_msg = f"API Error: {str(e)}"
+        if "Insufficient Balance" in str(e):
+            error_msg += "\n\nPlease check your DeepSeek API account balance."
+        return {"messages": [AIMessage(content=error_msg)]}
 
 def simple_grade_documents(state: AgentState):
     messages = state["messages"]
@@ -179,7 +201,7 @@ def generate(state: AgentState):
 
     headers = {
         "Accept": "application/json",
-        "Authorization": f"Bearer sk-1cddf19f9dc4466fa3ecea6fe10abec0",
+        "Authorization": f"Bearer {DEEPSEEK_API_KEY}",
         "Content-Type": "application/json"
     }
     
@@ -199,20 +221,23 @@ Focus on extracting and synthesizing the key findings from the research papers.
         "max_tokens": 1024
     }
     
-    print("Sending generate request to API...")
-    response = requests.post(
-        "https://api.deepseek.com/v1/chat/completions",
-        headers=headers,
-        json=data,
-        verify=False
-    )
-    
-    if response.status_code == 200:
+    try:
+        print("Sending generate request to API...")
+        response = requests.post(
+            "https://api.deepseek.com/v1/chat/completions",
+            headers=headers,
+            json=data,
+            verify=False,
+            timeout=30
+        )
+        response.raise_for_status()
+        
         response_text = response.json()['choices'][0]['message']['content']
         print("Final Answer:", response_text)
         return {"messages": [AIMessage(content=response_text)]}
-    else:
-        raise Exception(f"API call failed: {response.text}")
+    except Exception as e:
+        error_msg = f"Generation Error: {str(e)}"
+        return {"messages": [AIMessage(content=error_msg)]}
 
 def rewrite(state: AgentState):
     print("---REWRITE QUESTION---")
@@ -221,7 +246,7 @@ def rewrite(state: AgentState):
 
     headers = {
         "Accept": "application/json",
-        "Authorization": f"Bearer sk-1cddf19f9dc4466fa3ecea6fe10abec0",
+        "Authorization": f"Bearer {DEEPSEEK_API_KEY}",
         "Content-Type": "application/json"
     }
     
@@ -235,23 +260,23 @@ def rewrite(state: AgentState):
         "max_tokens": 1024
     }
     
-    print("Sending rewrite request...")
-    response = requests.post(
-        "https://api.deepseek.com/v1/chat/completions",
-        headers=headers,
-        json=data,
-        verify=False
-    )
-    
-    print("Status Code:", response.status_code)
-    print("Response:", response.text)
-    
-    if response.status_code == 200:
+    try:
+        print("Sending rewrite request...")
+        response = requests.post(
+            "https://api.deepseek.com/v1/chat/completions",
+            headers=headers,
+            json=data,
+            verify=False,
+            timeout=30
+        )
+        response.raise_for_status()
+        
         response_text = response.json()['choices'][0]['message']['content']
         print("Rewritten question:", response_text)
         return {"messages": [AIMessage(content=response_text)]}
-    else:
-        raise Exception(f"API call failed: {response.text}")
+    except Exception as e:
+        error_msg = f"Rewrite Error: {str(e)}"
+        return {"messages": [AIMessage(content=error_msg)]}
 
 tools_pattern = re.compile(r"Action: .*")
 
@@ -340,13 +365,8 @@ def main():
     }
     
     .stButton > button {
-        width: 100%;
-        margin-top: 20px;
         background-color: #4CAF50;
         color: white;
-        border: none;
-        padding: 10px 24px;
-        border-radius: 4px;
         transition: all 0.3s;
     }
     
@@ -356,14 +376,7 @@ def main():
     }
     
     .data-box {
-        padding: 20px;
-        border-radius: 10px;
-        margin: 10px 0;
         background-color: #2d2d2d;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.2);
-    }
-    
-    .research-box {
         border-left: 5px solid #2196F3;
     }
     
@@ -374,53 +387,19 @@ def main():
     .st-expander {
         background-color: #2d2d2d;
         border: 1px solid #3d3d3d;
-        border-radius: 4px;
-    }
-    
-    .st-expander label {
-        color: #ffffff !important;
-    }
-    
-    .stAlert {
-        background-color: #2d2d2d !important;
-    }
-    
-    .st-bb {
-        color: white !important;
-    }
-    
-    h1, h2, h3, h4, h5, h6 {
-        color: #ffffff !important;
-    }
-    
-    .stMarkdown {
-        color: #e0e0e0;
-    }
-    
-    .stWarning {
-        background-color: #423a2d !important;
     }
     </style>
     """, unsafe_allow_html=True)
 
     with st.sidebar:
-        st.markdown("""
-        <style>
-        .sidebar .sidebar-content {
-            background-color: #2d2d2d;
-        }
-        </style>
-        """, unsafe_allow_html=True)
-        
         st.header("📚 Available Data")
-        
         st.subheader("Research Database")
         for text in research_texts:
-            st.markdown(f'<div class="data-box research-box">{text}</div>', unsafe_allow_html=True)
+            st.markdown(f'<div class="data-box research-box" style="padding: 15px; margin: 10px 0; border-radius: 5px;">{text}</div>', unsafe_allow_html=True)
             
         st.subheader("Development Database")
         for text in development_texts:
-            st.markdown(f'<div class="data-box dev-box">{text}</div>', unsafe_allow_html=True)
+            st.markdown(f'<div class="data-box dev-box" style="padding: 15px; margin: 10px 0; border-radius: 5px;">{text}</div>', unsafe_allow_html=True)
 
     st.title("🤖 AI Research & Development Assistant")
     st.markdown("---")
@@ -431,36 +410,53 @@ def main():
     with col1:
         if st.button("🔍 Get Answer", use_container_width=True):
             if query:
-                with st.spinner('Processing your question...'):
-                    events = process_question(query, app, {"configurable": {"thread_id": "1"}})
-                    
-                    for event in events:
-                        if 'agent' in event:
-                            with st.expander("🔄 Processing Step", expanded=True):
-                                content = event['agent']['messages'][0].content
-                                if "Results:" in content:
-                                    st.markdown("### 📑 Retrieved Documents:")
-                                    docs_start = content.find("Results:")
-                                    docs = content[docs_start:]
-                                    st.info(docs)
-                        elif 'generate' in event:
-                            st.markdown("### ✨ Final Answer:")
-                            st.success(event['generate']['messages'][0].content)
+                try:
+                    with st.spinner('Processing your question...'):
+                        events = process_question(query, app, {"configurable": {"thread_id": "1"}})
+                        
+                        for event in events:
+                            if 'agent' in event:
+                                with st.expander("🔄 Processing Step", expanded=True):
+                                    content = event['agent']['messages'][0].content
+                                    if "Error" in content:
+                                        st.error(content)
+                                    elif "Results:" in content:
+                                        st.markdown("### 📑 Retrieved Documents:")
+                                        docs_start = content.find("Results:")
+                                        docs = content[docs_start:]
+                                        st.info(docs)
+                            elif 'generate' in event:
+                                content = event['generate']['messages'][0].content
+                                if "Error" in content:
+                                    st.error(content)
+                                else:
+                                    st.markdown("### ✨ Final Answer:")
+                                    st.success(content)
+                except Exception as e:
+                    st.error(f"""
+                    **Processing Error**  
+                    {str(e)}  
+                    Please check:  
+                    - API key configuration  
+                    - Account balance  
+                    - Network connection
+                    """)
             else:
                 st.warning("⚠️ Please enter a question first!")
 
     with col2:
         st.markdown("""
         ### 🎯 How to Use
-        1. Type your question in the text box
-        2. Click "Get Answer" to process
-        3. View retrieved documents and final answer
-        
+        1. Enter your question in the text box
+        2. Click the search button
+        3. Review processing steps
+        4. See final answer
+
         ### 💡 Example Questions
-        - What are the latest advancements in AI research?
-        - What is the status of Project A?
-        - What are the current trends in machine learning?
-        """, unsafe_allow_html=True)
+        - What's new in AI image recognition?
+        - How is Project B progressing?
+        - Recent machine learning trends?
+        """)
 
 if __name__ == "__main__":
     main()