Update app.py

app.py

@@ -335,10 +335,10 @@ def fine_tune_report(adjustment_request: str, openai_api_key: str, serpapi_api_k
     logging.info("fine_tune_report: Fine-tuning process completed.")
     return updated_report_html, updated_qa
 
-def suggest_improvements(report_html: str) -> str:
-    """
-    Generate 10 improvement proposals in the required numbered format.
-    """
+def suggest_improvements(report_html: str, openai_api_key: str, serpapi_api_key: str) -> str:
+    os.environ["OPENAI_API_KEY"] = openai_api_key
+    os.environ["SERPAPI_API_KEY"] = serpapi_api_key
+
     prompt = (
         "You are a technical editor. Based on the following full HTML report, generate exactly 10 proposed improvement suggestions. "
         "Format each proposal as a numbered list item in the following style:\n"
@@ -2953,7 +2953,7 @@ def main():
 
         suggest_improvements_button.click(
             fn=suggest_improvements,
-            inputs=[final_report],
+            inputs=[final_report, openai_api_key_input, serpapi_api_key_input],
             outputs=[adjustmentguidelines]
         )