Update app.py

app.py

@@ -34,9 +34,6 @@ def get_data(url):
             "List me all grants or funds, short summary of grant description, "
             "the organisations funding them, the value of the grant as an integer, "
             "the due date, eligible countries, sector and eligibility criteria for applicants."
-
-            "if you cant find grants return any useful information you can find"
-            
         ),
         source=url,
         config=graph_config,
@@ -74,9 +71,6 @@ def process_multiple_urls(urls):
             result = get_data(url)
             if result and "grants" in result:
                 all_data["grants"].extend(result["grants"])
-            else:
-                all_data = result
-                
         except Exception as e:
             st.error(f"⚠️ Error processing URL: {url} - {str(e)}")
             continue
@@ -87,29 +81,16 @@ def process_multiple_urls(urls):
 
 
 def convert_to_csv(data):
-    try:
-        df = pd.DataFrame(data["grants"])
-        return df.to_csv(index=False).encode("utf-8")
-    except:
-        df = pd.DataFrame(data)
-        return df.to_csv(index=False).encode("utf-8")
+    df = pd.DataFrame(data["grants"])
+    return df.to_csv(index=False).encode("utf-8")
 
 
 def convert_to_excel(data):
-    try:
-        df = pd.DataFrame(data["grants"])
-        buffer = io.BytesIO()
-        with pd.ExcelWriter(buffer, engine="xlsxwriter") as writer:
-            df.to_excel(writer, sheet_name="Grants", index=False)
-        return buffer.getvalue()
-
-    except:
-        df = pd.DataFrame(data)
-        buffer = io.BytesIO()
-        with pd.ExcelWriter(buffer, engine="xlsxwriter") as writer:
-            df.to_excel(writer, sheet_name="Grants", index=False)
-        return buffer.getvalue()
-            
+    df = pd.DataFrame(data["grants"])
+    buffer = io.BytesIO()
+    with pd.ExcelWriter(buffer, engine="xlsxwriter") as writer:
+        df.to_excel(writer, sheet_name="Grants", index=False)
+    return buffer.getvalue()
 
 
 def create_knowledge_base(data):
@@ -203,10 +184,7 @@ def main():
 
         # Data Preview and Download Options in Main Panel
         with st.expander(f"📊 Preview Grant Data {len(st.session_state.scraped_data['grants'])} grants"):
-            try:
-                st.dataframe(st.session_state.scraped_data["grants"])
-            except:
-                st.dataframe(st.session_state.scraped_data)
+            st.dataframe(st.session_state.scraped_data["grants"])
 
         col1, col2, col3 = st.columns([1, 1, 2]) # Adjust column widths for better layout
 
@@ -252,12 +230,9 @@ def main():
 
             query = st.text_input("Your question:", key="chat_input")
             if query:
-                if st.session_state.qa_chain:
-                    with st.spinner("Generating response..."):
-                        response = chat_with_knowledge_base(query)
-                        st.session_state.chat_history.append({"query": query, "response": response["answer"]})
-                else:
-                    st.error("Knowledge base not initialized. Please load data as knowledge base.")
+                with st.spinner("Generating response..."):
+                    response = chat_with_knowledge_base(query)
+                    st.session_state.chat_history.append({"query": query, "response": response["answer"]})
 
             if st.session_state.chat_history:
                 st.subheader("Chat History")
@@ -279,6 +254,5 @@ def main():
         unsafe_allow_html=True,
     )
 
-
 if __name__ == "__main__":
     main()