Update app.py

app.py

@@ -12,25 +12,44 @@ FLOW_ID = "0853dfd7-558c-4958-9ca4-dc9ca8c69302"
 APPLICATION_TOKEN = os.environ.get("APP_TOKEN")
 ENDPOINT = "materialssciencebot"
 
-def run_flow(message: str) -> dict:
+# Function to call API with improved error handling
+def run_flow(message: str):
     api_url = f"{BASE_API_URL}/lf/{LANGFLOW_ID}/api/v1/run/{ENDPOINT}"
-
+    
     payload = {
         "input_value": message,
         "output_type": "chat",
         "input_type": "chat",
     }
-   
-    headers = {"Authorization": "Bearer " + APPLICATION_TOKEN, "Content-Type": "application/json"}
-    response = requests.post(api_url, json=payload, headers=headers)
-    return response.json()
+
+    headers = {"Authorization": f"Bearer {APPLICATION_TOKEN}", "Content-Type": "application/json"}
+    
+    try:
+        response = requests.post(api_url, json=payload, headers=headers, timeout=60, stream=True)
+        response.raise_for_status()  # Raise error for HTTP failures
+        
+        collected_response = ""
+        for chunk in response.iter_content(chunk_size=1024):
+            if chunk:
+                collected_response += chunk.decode()
+                yield collected_response  # Stream output gradually
+        
+    except requests.exceptions.RequestException as e:
+        yield f"Error: {str(e)}"
+    except json.JSONDecodeError:
+        yield "Error: Invalid JSON response from API"
+
+# Cache previous API responses to improve performance
+@st.cache_data
+def get_response(message: str):
+    return list(run_flow(message))[-1]  # Get final response
 
 st.markdown("""
     <style>
         .footer {
             position: fixed;
             bottom: 0;
-            right: 0;
+            left: 0;
             padding: 10px;
             font-size: 16px;
             color: #333;
@@ -45,21 +64,30 @@ st.markdown("""
 def main():
     st.title("Materials Science Bot")
     st.markdown("<h4 style='font-size: 20px;'> Ask anything related to the world of materials! 😉</h4>", unsafe_allow_html=True)
-    type = st.text_area("Type", placeholder="What is oxidation?...")
+    message = st.text_area("Message", placeholder="What is oxidation?...")
 
     if st.button("Run"):
-        if not type.strip():
-            st.error("Please enter a query")
+        if not message.strip():
+            st.error("Please enter a message")
             return
         
         try:
             with st.spinner("Running flow..."):
-                response = run_flow(type)
+                response_placeholder = st.empty()
+                full_response = ""
+
+                for chunk in run_flow(message):  # Stream response
+                    full_response += chunk  # Append the latest chunk
+                    try:
+                        json_response = json.loads(full_response)  # Convert to JSON
+                        extracted_text = json_response["outputs"][0]["outputs"][0]["results"]["message"]["text"]
+                        response_placeholder.markdown(extracted_text)  # Display only AI response
+                    except (json.JSONDecodeError, KeyError):
+                        response_placeholder.markdown(full_response)  # Show raw response if parsing fails
 
-            response = response['outputs'][0]['outputs'][0]['results']['type']['text']
-            st.markdown(response)
         except Exception as e:
-            st.error(str(e))
+            st.error(f"⚠️ Error: {str(e)}")
+
 
 if __name__ == "__main__":
     main()