updated pythhon file locations

app.py

@@ -62,6 +62,10 @@ st.markdown("This is an Agent which can do Market Analysis and Generate Use Case
 name = st.text_input("Enter Name of the company:", placeholder="e.g., Google, Apple, Nike")
 link = st.text_input("Enter the company's link:", placeholder="e.g., https://www.google.com, https://www.apple.com, https://www.nike.com")
 
+# Initialize session state if not present
+if 'article_text' not in st.session_state:
+    st.session_state.article_text = ""
+
 # Button to generate article
 if st.button("Generate Article"):
     # Check if API keys are provided
@@ -93,45 +97,54 @@ if st.button("Generate Article"):
         
         # Extract the article text from the result
         if isinstance(result, str):
-            article_text = result
+            st.session_state.article_text = result
         elif isinstance(result, dict) and 'article' in result:
-            article_text = result['article']
+            st.session_state.article_text = result['article']
         else:
-            article_text = str(result)  
+            st.session_state.article_text = str(result)  
         
         # Display the article text
-        st.markdown(article_text)
+        st.markdown(st.session_state.article_text)
+
+# Create three columns for download buttons
+col1, col2, col3 = st.columns(3)
 
-        # Create three columns for download buttons
-        col1, col2, col3 = st.columns(3)
+# Download button for the article
+with col1:
+    if st.session_state.article_text:
+        st.download_button(
+            label="Download Article",
+            data=st.session_state.article_text,
+            file_name=f"{name.replace(' ', '_').lower()}_market_and_use_case_analysis.txt",
+            mime="text/plain"
+        )
+    else:
+        st.write("Article not generated yet.")
 
-        # Download button for the article
-        with col1:
+# Download button for ideas
+with col2:
+    if st.session_state.article_text:
+        with open("output/ideas.md", "rb") as fp:
             st.download_button(
-                label="Download Article",
-                data=article_text,
-                file_name=f"{name.replace(' ', '_').lower()}_market_and_use_case_analysis.txt",
+                label="Download Ideas",
+                data=fp,
+                file_name=f"{name.replace(' ', '_').lower()}_ideas.txt",
                 mime="text/plain"
             )
-
-        # Download button for ideas
-        with col2:
-            with open("output/ideas.md", "rb") as fp:
-                st.download_button(
-                    label="Download Ideas",
-                    data=fp,
-                    file_name=f"{name.replace(' ', '_').lower()}_ideas.txt",
-                    mime="text/plain"
+    else:
+        st.wrtie("Ideas are ideating")
+# Download button for resources
+with col3:
+    if st.session_state.article_text: 
+        with open("output/resouce.md", "rb") as fp:
+            st.download_button(
+                label="Download Resources",
+                data=fp,
+                file_name=f"{name.replace(' ', '_').lower()}_resources.txt",
+                mime="text/plain"
             )
-
-        # Download button for resources
-        with col3:
-            with open("output/resouce.md", "rb") as fp:
-                 st.download_button(
-                    label="Download Ideas",
-                    data=fp,
-                    file_name=f"{name.replace(' ', '_').lower()}_ideas.txt",
-                    mime="image/txt"
-    )
+    else:
+        st.write("Resources are being researched")
 
 st.markdown("---------")
+

src/research_agent/crew.py

@@ -1,7 +1,7 @@
 import sys
 from crewai import Agent, Crew, Process, Task, LLM
 from crewai.project import CrewBase, agent, crew, task
-from research_agent.tools.tool import search_tool, website_search_tool,pdf_search_tool
+from src.research_agent.tools.tool import search_tool, website_search_tool,pdf_search_tool
 import streamlit as st
 import os
 sys.path.append('..')