Update app.py

app.py

@@ -141,4 +141,62 @@ class EmailAgent:
         
         data = {
             "messages": [{"role": "user", "content": prompt}],
-            "model": "llama3-8b
+            "model": "llama3-8b-8192"
+        }
+        
+        response = requests.post(url, headers=headers, json=data)
+        if response.status_code == 200:
+            return response.json()["choices"][0]["message"]["content"].strip()
+        else:
+            print(f"Error: {response.status_code}, {response.text}")
+            return "Error generating email. Please check your API key or try again later."
+
+    # Main loop following ReAct pattern
+    def run(self):
+        self.reason_about_data()  # Reasoning step
+        self.fetch_linkedin_data()  # Fetch LinkedIn data
+        self.fetch_company_info()  # Fetch company data
+        # Scrape the company's website for role-specific information or use defaults
+        if not self.scrape_role_from_website():
+            self.use_default_role_description()  
+        # Reflect on whether the data is sufficient
+        if self.reflect_on_data():
+            return self.generate_email()  # Final action: generate email
+        else:
+            return "Error: Not enough data to generate the email."
+
+# Define the Gradio interface and the main app logic
+def gradio_ui():
+    # Input fields
+    name_input = gr.Textbox(label="Your Name", placeholder="Enter your name")
+    company_input = gr.Textbox(label="Company Name or URL", placeholder="Enter the company name or website URL")
+    role_input = gr.Textbox(label="Role Applying For", placeholder="Enter the role you are applying for")
+    email_input = gr.Textbox(label="Your Email Address", placeholder="Enter your email address")
+    phone_input = gr.Textbox(label="Your Phone Number", placeholder="Enter your phone number")
+    linkedin_input = gr.Textbox(label="Your LinkedIn URL", placeholder="Enter your LinkedIn profile URL")
+    word_limit_slider = gr.Slider(minimum=50, maximum=300, step=10, label="Email Word Limit", value=150)  # Word limit slider
+    
+    # Output field
+    email_output = gr.Textbox(label="Generated Email", placeholder="Your generated email will appear here", lines=10)
+
+    # Function to create and run the email agent
+    def create_email(name, company_name, role, email, phone, linkedin_url, word_limit):
+        agent = EmailAgent(linkedin_url, company_name, role, word_limit, name, email, phone, linkedin_url)
+        return agent.run()
+
+    # Gradio interface
+    demo = gr.Interface(
+        fn=create_email,
+        inputs=[name_input, company_input, role_input, email_input, phone_input, linkedin_input, word_limit_slider],
+        outputs=[email_output],
+        title="Email Writing AI Agent with ReAct",
+        description="Generate a professional email for a job application using LinkedIn data, company info, and role description.",
+        allow_flagging="never"
+    )
+    
+    # Launch the Gradio app
+    demo.launch()
+
+# Start the Gradio app when running the script
+if __name__ == "__main__":
+    gradio_ui()