Update app.py

app.py

@@ -249,103 +249,17 @@ def stream_pitchdeck(company_name, url):
 
 import gradio as gr
 
-def get_links_user_prompt(website):
-    user_prompt = f"Here is the list of links on the website of {website.url} - "
-    user_prompt += "please decide which of these are relevant web links for a brochure about the company, respond with the full https URL in JSON format. \
-Do not include Terms of Service, Privacy, email links.\n"
-    user_prompt += "Links (some might be relative links):\n"
-    user_prompt += "\n".join(website.links)
-    return user_prompt
-
-system_prompt = """
-You are a business analyst specializing in creating compelling pitch decks from company website content. Respond in markdown.\
-Your task is to analyze multiple relevant pages from a company's website and synthesize the information into a concise, \
-professional pitch deck that the company can use for business development, investor relations.
-
-Your pitch deck should be formatted in markdown and include the following sections when information is available:
-- Company overview and mission
-- Products/services and value proposition
-- Target market and customer base
-- Company culture and values
-- Growth potential and achievements
-- Contact information
-- End with We are always happy to help you with (list the services again)
-
-Focus on creating a narrative that highlights the company's strengths, unique positioning, and opportunities for partnership,\
- investment. Ensure the content is professional, engaging, and suitable for multiple audiences including potential clients and investors."""
-
-def get_all_details(url):
-    result = "Landing page:\n"
-    result += Website(url).get_contents()
-    links = get_links(url)
-    print("Found links:", links)
-    for link in links.get("links", []):
-        result += f"\n\n{link.get('type', 'Unknown Type')}\n"
-        # Ensure the URL is valid before attempting to scrape
-        if link.get('url') and (link['url'].startswith('http://') or link['url'].startswith('https://')):
-            try:
-                result += Website(link["url"]).get_contents()
-            except Exception as e:
-                result += f"Could not retrieve content for {link['url']}: {e}\n"
-        else:
-            result += f"Invalid URL found: {link.get('url')}\n"
-    return result
-
-def get_pitch_user_prompt(company_name, url):
-    user_prompt = f"You are looking at a company called: {company_name}\n"
-    user_prompt += f"Here are the contents of its landing page and other relevant pages; use this information to build a short pitch deck for the company, of the company in markdown.\n"
-    try:
-        user_prompt += get_all_details(url)
-    except Exception as e:
-        user_prompt += f"Error getting website details: {e}\n"
-        print(f"Error getting website details for {url}: {e}")
-        # Provide a fallback or stop processing if website cannot be scraped
-        return "Could not retrieve website details to generate pitch deck."
-
-    user_prompt = user_prompt[:5_000] # Truncate if more than 5,000 characters
-    return user_prompt
-
-
-def generate_pitch_deck(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:
-        response = openai.chat.completions.create(
-            model=MODEL,
-            messages=[
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": user_prompt}
-              ],
-            stream=True  # Enable streaming for better UX in Gradio
-        )
-
-        # Stream the response
-    response = ""
-    display_handle = display(Markdown(""), display_id=True)
-    for chunk in stream:
-        response += chunk.choices[0].delta.content or ''
-        response = response.replace("```","").replace("markdown", "")
-        update_display(Markdown(response), display_id=display_handle.display_id) # Yield intermediate results for streaming
-
-        return response # Return the final result
-    except Exception as e:
-        return f"An error occurred during OpenAI API call: {e}"
-
+reate the Gradio interface
 iface = gr.Interface(
-    fn=generate_pitch_deck,
+    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 their website content."
+    description="Enter the company name and website URL to generate a pitch deck based on the website content."
 )
 
-iface.launch()
+# Launch the Gradio app
+iface.launch(d)