Update app.py

app.py

@@ -245,13 +245,40 @@ def stream_pitchdeck(company_name, url):
 
 
 #gradio
-
-
 import gradio as gr
 
-#create the Gradio interface
-iface = gr.Interface(
-    fn=stream_pitchdeck,
+def stream_pitchdeck2(company_name, url):
+    if not company_name or not url:
+        return "Please provide both company name and URL."
+    
+    user_prompt = get_pitch_user_prompt(company_name, url)
+    
+    if user_prompt == "Could not retrieve website details to generate pitch deck.":
+        return user_prompt
+    
+    try:
+        stream = openai.chat.completions.create(
+            model=MODEL,
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": user_prompt}
+            ],
+            stream=True
+        )
+
+        response = ""
+        for chunk in stream:
+            response += chunk.choices[0].delta.content or ''
+            response = response.replace("```","").replace("markdown", "")
+            yield response  # Use yield for Gradio streaming
+        
+        return response
+    except Exception as e:
+        return f"An error occurred during OpenAI API call: {e}"
+
+# Create the Gradio interface
+iface2 = gr.Interface(
+    fn=stream_pitchdeck2,
     inputs=[
         gr.Textbox(label="Company Name"),
         gr.Textbox(label="Company Website URL")
@@ -260,6 +287,21 @@ iface = gr.Interface(
     title="Company Pitch Deck Generator",
     description="Enter the company name and website URL to generate a pitch deck based on the website content."
 )
-
-# Launch the Gradio app
-iface.launch()
+iface2.launch(debug = True, share=True)
+
+# import gradio as gr
+
+# #create the Gradio interface
+# iface = gr.Interface(
+#     fn=stream_pitchdeck,
+#     inputs=[
+#         gr.Textbox(label="Company Name"),
+#         gr.Textbox(label="Company Website URL")
+#     ],
+#     outputs=gr.Markdown(label="Generated Pitch Deck"),
+#     title="Company Pitch Deck Generator",
+#     description="Enter the company name and website URL to generate a pitch deck based on the website content."
+# )
+
+# # Launch the Gradio app
+# iface.launch()