Update app.py

app.py

@@ -7,6 +7,87 @@ import time
 from PIL import Image
 from streamlit_image_comparison import image_comparison
 import numpy as np
+#import chromadb
+
+from textwrap import dedent
+import google.generativeai as genai
+
+api_key = os.environ["OPENAI_API_KEY"]
+
+from openai import OpenAI
+import numpy as np
+# Assuming chromadb and TruLens are correctly installed and configured
+
+#from chromadb.utils.embedding_functions import
+
+# Google Langchain
+from langchain_google_genai import GoogleGenerativeAI
+
+#Crew imports
+from crewai import Agent, Task, Crew, Process
+
+# Retrieve API Key from Environment Variable
+GOOGLE_AI_STUDIO = os.environ.get('GOOGLE_API_KEY')
+
+# Ensure the API key is available
+if not GOOGLE_AI_STUDIO:
+    raise ValueError("API key not found. Please set the GOOGLE_AI_STUDIO2 environment variable.")
+
+# Set gemini_llm
+gemini_llm = GoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_AI_STUDIO)
+
+
+
+# CrewAI ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+
+def crewai_process_gemini(research_topic):
+    # Define your agents with roles and goals
+    GeminiAgent = Agent(
+        role='Story Writer',
+        goal='To create a story from bullet points.',
+        backstory="""You are an expert writer that understands how to make the average extraordinary on paper """,
+        verbose=True,
+        allow_delegation=False,
+        llm = gemini_llm,
+        tools=[
+                GeminiSearchTools.gemini_search
+                   
+      ]
+
+    )
+
+
+    # Create tasks for your agents
+    task1 = Task(
+        description=f"""From {research_topic} create your story by writing at least one sentence about each bullet point 
+        and make sure you have a transitional statement between scenes . BE VERBOSE.""",
+        agent=GeminiAgent
+    )
+
+    # Instantiate your crew with a sequential process
+    crew = Crew(
+        agents=[GeminiAgent],
+        tasks=[task1],
+        verbose=2,
+        process=Process.sequential
+    )
+
+    # Get your crew to work!
+    result = crew.kickoff()
+    
+    return result
+
+
+
+
+# Tool import
+from crewai.tools.gemini_tools import GeminiSearchTools
+from crewai.tools.mixtral_tools import MixtralSearchTools
+from crewai.tools.zephyr_tools import ZephyrSearchTools
+from crewai.tools.phi2_tools import 
+
+
 
 
 
@@ -165,9 +246,32 @@ with tab2:
         
         st.header("Agent Advisors")
         st.write("Would you like to optimize your HIN number?")
-        
-        st.header("My Incentive")
-        st.write("Total incentive for HIN optimization")
+
+        # Selection box for the function to execute
+        process_selection = st.selectbox(
+            'Choose the process to run:',
+            ('crewai_process_gemini', 'crewai_process_mixtral_crazy', 'crewai_process_mixtral_normal', 'crewai_process_zephyr_normal', 'crewai_process_phi2')
+        )
+    
+        # Button to execute the chosen function
+        if st.button('Run Process'):
+            if research_topic:  # Ensure there's a topic provided
+                if process_selection == 'crewai_process_gemini':
+                    result = crewai_process_gemini(research_topic)
+                elif process_selection == 'crewai_process_mixtral_crazy':
+                    result = crewai_process_mixtral_crazy(research_topic)
+                elif process_selection == 'crewai_process_mixtral_normal':
+                    result = crewai_process_mixtral_normal(research_topic)
+                elif process_selection == 'crewai_process_zephyr_normal':
+                    result = crewai_process_zephyr_normal(research_topic)
+                elif process_selection == 'crewai_process_phi2':
+                    result = crewai_process_phi2(research_topic)
+                st.write(result)
+            else:
+                st.warning('Please enter a research topic.')
+            
+            st.header("My Incentive")
+            st.write("Total incentive for HIN optimization")
     
     with col2:
         st.header("Green Smart Village Layout")