alpha version 1.0

.gitignore

@@ -0,0 +1 @@
+.env

app.py

@@ -3,148 +3,149 @@ import sys
 import streamlit as st
 from src.research_agent.crew import MarketUseCaseCrew
 import streamlit as st
-from crewai import Crew, Process
 import os
+from dotenv import load_dotenv
+load_dotenv()
 sys.path.append('..')
-# Configure the Streamlit page
-st.set_page_config(page_title="CrewAI Article Generator", page_icon="📝", layout="wide")
 
-# Custom CSS for styling the page
-st.markdown("""
-<style>
-    .reportview-container {
-        background: #f0f2f6
-    }
-    .main {
-        background: #00000;
-        padding: 3rem;
-        border-radius: 10px;
-        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-    }
-    .stButton>button {
-        background-color: #0000;
-        color: white;
-        border-radius: 5px;
-    }
-    .stTextInput>div>div>input {
-        border-radius: 5px;
-    }
-    .sidebar .sidebar-content {
-        background-color: #f8f9fa;
-    }
-</style>
-""", unsafe_allow_html=True)
 
-# Sidebar for API key input
-st.sidebar.title("📊 API Configuration")
-st.sidebar.markdown("Enter your API keys below:")
 
-# Input fields for API keys
-serper_api_key = st.sidebar.text_input("Serper API Key", type="password")
-gemini_api_key = st.sidebar.text_input("Gemini Flash API Key", type="password")
+class MarketUseCaseGen:
 
-# Button to save API keys
-if st.sidebar.button("Save API Keys"):
-    # Check if both API keys are provided
-    if serper_api_key and gemini_api_key:
-        # Set environment variables for API keys
-        os.environ['SERPER_API_KEY'] = serper_api_key # Installed the SERPER_API_KEY in the environment
-        os.environ['GOOGLE_API_KEY'] = gemini_api_key # Installed the gemini_api_key in the environment
-        st.sidebar.success("API keys saved successfully!")
-    else:
-        st.sidebar.error("Please enter both API keys.")
-
-# Main content section
-st.title("📝 Market and Research Analysis ")
-st.markdown("This is an Agent which can do Market Analysis and Generate Use Cases in the AI space for you")
-
-# Input fields for company name and website
-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")
+    def generation(self, name, link):
+        inputs = {"company": name, "company_link": link}
+        return MarketUseCaseCrew().crew().kickoff(inputs=inputs)
+    
+    def use_case_generation(self):
+        if st.session_state.generation:
+            with st.spinner(f"Researching and generating uses cases about AI..."):
+            # Set progress to 50%
+                progress_bar = st.progress(50)
+                st.session_state.article_text = self.generation(
+                    st.session_state.name, st.session_state.link
+                )
+                progress_bar.progress(100)
+                # Display the generated article
+                st.subheader("Generated Article:")
+        if st.session_state.article_text and st.session_state.article_text != "":
+            with st.container():
+            # Display the article text
+                st.markdown(st.session_state.article_text)
+                col1, col2, col3 = st.columns(3)
 
-# Initialize session state if not present
-if 'article_text' not in st.session_state:
-    st.session_state.article_text = ""
+                # 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"{st.session_state.name.replace(' ', '_').lower()}_market_and_use_case_analysis.txt",
+                            mime="text/plain"
+                        )
+                    else:
+                        st.write("Article not generated yet.")
 
-# Button to generate article
-if st.button("Generate Article"):
-    # Check if API keys are provided
-    if not serper_api_key or not gemini_api_key:
-        st.error("Please enter both API keys in the sidebar before generating.")
-    # Check if company name and website are provided
-    elif not name and link:
-        st.warning("Please enter the company name and website")
-    else:
-        # Create a progress bar
-        progress_bar = st.progress(0)
-        # Input data for the article generation
-        inputs = {
-            'company': name,
-            'company_link': link
-        }
+                # 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 Ideas",
+                                data=fp,
+                                file_name=f"{st.session_state.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"{st.session_state.name.replace(' ', '_').lower()}_resources.txt",
+                                mime="text/plain"
+                            )
+                    else:
+                        st.write("Resources are being researched")
+                st.markdown("---------")
+            st.session_state.generation=False
+    
+    def sidebar(self):
+        with st.sidebar:
+            st.title("📊 API Configuration")
+            st.markdown("Enter your API keys below:")
 
-        # Use the MarketUseCaseCrew class to generate the article
-        with st.spinner(f"Researching and generating uses cases about AI for '{name}'..."):
-            # Set progress to 50%
-            progress_bar.progress(50)
-            # Call the kickoff method to generate the article
-            result = MarketUseCaseCrew().crew().kickoff(inputs=inputs)
+            # Input fields for API keys
+            serper_api_key = st.text_input("Serper API Key", type="password")
+            gemini_api_key = st.text_input("Gemini Flash API Key", type="password")
 
-        # Set progress to 100%
-        progress_bar.progress(100)
-        # Display the generated article
-        st.subheader("Generated Article:")
-        
-        # Extract the article text from the result
-        if isinstance(result, str):
-            st.session_state.article_text = result
-        elif isinstance(result, dict) and 'article' in result:
-            st.session_state.article_text = result['article']
-        else:
-            st.session_state.article_text = str(result)  
-        
-        # Display the article text
-        st.markdown(st.session_state.article_text)
+            # Button to save API keys
+            if st.button("Save API Keys"):
+                # Check if both API keys are provided
+                if serper_api_key and gemini_api_key:
+                    # Set environment variables for API keys
+                    os.environ['SERPER_API_KEY'] = serper_api_key # Installed the SERPER_API_KEY in the environment
+                    os.environ['GOOGLE_API_KEY'] = gemini_api_key # Installed the gemini_api_key in the environment
+                    st.success("API keys saved successfully!")
+                else:
+                    st.error("Please enter both API keys.")
 
-# Create three columns for download buttons
-col1, col2, col3 = st.columns(3)
+            # Input fields for company name and website
+            name = st.text_input("Enter Name of the company:", key='name',placeholder="e.g., Google, Apple, Nike")
+            link = st.text_input("Enter the company's link:", key='link',placeholder="e.g., https://www.google.com, https://www.apple.com, https://www.nike.com")
 
-# 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.")
+            if st.button("Generate Article"):
+                # Check if API keys are provided
+                if not serper_api_key or not gemini_api_key:
+                    st.error("Please enter both API keys in the sidebar before generating.")
+                     # Check if company name and website are provided
+                elif not name and link:
+                    st.warning("Please enter the company name and website")
+                else:
+                    # Create a progress bar
+                    progress_bar = st.progress(0)
+                    st.session_state.generation=True
 
-# 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 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"
-            )
-    else:
-        st.write("Resources are being researched")
+    def render(self):
+        st.set_page_config(page_title="CrewAI Article Generator", page_icon="📝", layout="wide")
+        # Main content section
+        st.markdown("""
+            <style>
+                .reportview-container {
+                    background: #f0f2f6
+                }
+                .main {
+                    background: #00000;
+                    padding: 3rem;
+                    border-radius: 10px;
+                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+                }
+                .stButton>button {
+                    background-color: #0000;
+                    color: white;
+                    border-radius: 5px;
+                }
+                .stTextInput>div>div>input {
+                    border-radius: 5px;
+                }
+                .sidebar .sidebar-content {
+                    background-color: #f8f9fa;
+                }
+            </style>
+            """, unsafe_allow_html=True)
+        if "article_text" not in st.session_state:
+            st.session_state.article_text = ""
+        if "generation" not in st.session_state:
+            st.session_state.generation = False
+        if "name" not in st.session_state:
+            st.session_state.website = ""
+        if "link" not in st.session_state:
+            st.session_state.link = ""
 
-st.markdown("---------")
+        self.sidebar()
+        self.use_case_generation()
 
+if __name__ == "__main__":
+    MarketUseCaseGen().render()

src/research_agent/crew.py

@@ -4,46 +4,20 @@ from crewai.project import CrewBase, agent, crew, task
 from src.research_agent.tools.tool import search_tool, website_search_tool,pdf_search_tool
 import streamlit as st
 import os
+from dotenv import load_dotenv
+load_dotenv()
 sys.path.append('..')
-def streamlit_callback(step_output):
-	"""Callback function to display step output in Streamlit."""
-	st.markdown("---")
-	for step in step_output:
-		if isinstance(step, tuple) and len(step) == 2:
-			action, observation = step
-			if isinstance(action, dict) and "tool" in action and "tool_input" in action and "log" in action:
-				st.markdown(f"# Action")
-				st.markdown(f"**Tool:** {action['tool']}")
-				st.markdown(f"**Tool Input:** {action['tool_input']}")
-				st.markdown(f"**Log:** {action['log']}")
-				if 'Action' in action:  # Check if 'Action' key exists before using it
-					st.markdown(f"**Action:** {action['Action']}")
-					st.markdown(f"**Action Input:** ```json\n{action['tool_input']}\n```")
-			elif isinstance(action, str):
-				st.markdown(f"**Action:** {action}")
-			else:
-				st.markdown(f"**Action:** {str(action)}")
-
-			st.markdown(f"**Observation**")
-			if isinstance(observation, str):
-				observation_lines = observation.split('\n')
-				for line in observation_lines:
-					st.markdown(line)
-			else:
-				st.markdown(str(observation))
-		else:
-			st.markdown(step)
 
 @CrewBase
 class MarketUseCaseCrew:
 	def llm(self):
-		return LLM(model="gemini/gemini-1.5-flash-002", temperature=0.01, api_key=os.environ["GOOGLE_API_KEY"])
+		return LLM(model="gemini/gemini-1.5-flash-002", api_key=os.getenv("GOOGLE_API_KEY"))
 	
 	@agent
 	def researcher(self) -> Agent:
 		return Agent(
 			config=self.agents_config['researcher'],
-			tools=[search_tool,website_search_tool], # Example of custom tool, loaded on the beginning of file
+			tools=[search_tool,website_search_tool, pdf_search_tool], # Example of custom tool, loaded on the beginning of file
 			verbose=True,
 			llm=self.llm(),
 			allow_delegation=True,