Update app.py

app.py

@@ -32,6 +32,9 @@ elif model_choice == "OpenAI":
     else:
         llm = ChatOpenAI(api_key=openai_api_key, model="gpt-4o")
 
+# Model Selection on Main Screen
+st.header("Model Selection")
+model_choice = st.selectbox("Select LLM Provider", ["Groq-based LLM", "OpenAI"])
 
 # Title and Application Introduction
 st.title("Patent Strategy and Innovation Consultant")
@@ -252,41 +255,49 @@ def display_table(analyst_output):
 
 # Main Execution Block
 if st.button("Generate Insights"):
-    with st.spinner('Processing...'):
-        try:
-            start_time = time.time()
-            results = crew.kickoff(inputs={"topic": patent_area, "stakeholder": stakeholder})
-            elapsed_time = time.time() - start_time
-
-            writer_output = getattr(results.tasks_output[2], "raw", "No details available.")
-            if writer_output:
-                st.markdown("### Final Report")
-                st.write(writer_output)
-            else:
-                st.warning("No final report available.")
-
-            with st.expander("Explore Detailed Insights"):
-                tab1, tab2 = st.tabs(["Planner's Insights", "Analyst's Analysis"])
-
-                with tab1:
-                    planner_output = getattr(results.tasks_output[0], "raw", "No details available.")
-                    st.write(planner_output)
-
-                with tab2:
-                    analyst_output = getattr(results.tasks_output[1], "raw", "No details available.")
-                    st.write(analyst_output)
-
-                    charts = []
-                    if enable_advanced_analysis:
-                        charts = create_visualizations(analyst_output)
-
-                    table_data = display_table(analyst_output)
-
-            st.success(f"Analysis completed in {elapsed_time:.2f} seconds.")
-            pdf_path = generate_pdf_report(writer_output, charts=charts, table_data=table_data, metadata={"Technology Area": patent_area, "Stakeholder": stakeholder})
-            with open(pdf_path, "rb") as report_file:
-                st.download_button("Download Report", data=report_file, file_name="Patent_Strategy_Report.pdf")
-
-        except Exception as e:
-            logging.error(f"An error occurred during execution: {e}")
-            st.error(f"An error occurred during execution: {e}")
+    if llm is None:
+        st.error("Cannot proceed without a valid API key for the selected model.")
+    else:
+        with st.spinner('Processing...'):
+            try:
+                start_time = time.time()
+                results = crew.kickoff(inputs={"topic": patent_area, "stakeholder": stakeholder})
+                elapsed_time = time.time() - start_time
+
+                writer_output = getattr(results.tasks_output[2], "raw", "No details available.")
+                if writer_output:
+                    st.markdown("### Final Report")
+                    st.write(writer_output)
+                else:
+                    st.warning("No final report available.")
+
+                with st.expander("Explore Detailed Insights"):
+                    tab1, tab2 = st.tabs(["Planner's Insights", "Analyst's Analysis"])
+
+                    with tab1:
+                        planner_output = getattr(results.tasks_output[0], "raw", "No details available.")
+                        st.write(planner_output)
+
+                    with tab2:
+                        analyst_output = getattr(results.tasks_output[1], "raw", "No details available.")
+                        st.write(analyst_output)
+
+                        charts = []
+                        if enable_advanced_analysis:
+                            charts = create_visualizations(analyst_output)
+
+                        table_data = display_table(analyst_output)
+
+                st.success(f"Analysis completed in {elapsed_time:.2f} seconds.")
+                pdf_path = generate_pdf_report(
+                    writer_output,
+                    charts=charts,
+                    table_data=table_data,
+                    metadata={"Technology Area": patent_area, "Stakeholder": stakeholder}
+                )
+                with open(pdf_path, "rb") as report_file:
+                    st.download_button("Download Report", data=report_file, file_name="Patent_Strategy_Report.pdf")
+
+            except Exception as e:
+                logging.error(f"An error occurred during execution: {e}")
+                st.error(f"An error occurred during execution: {e}")