Update app.py

app.py

@@ -2,12 +2,11 @@ import os
 import gradio as gr
 from groq import Groq
 from dotenv import load_dotenv
-from datetime import datetime
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
 from reportlab.lib.styles import getSampleStyleSheet
 from reportlab.lib.units import inch
 
-# Load environment variables
+# Load API Key
 load_dotenv()
 GROQ_API_KEY = os.getenv("Freelancing_key")
 
@@ -17,10 +16,10 @@ client = Groq(api_key=GROQ_API_KEY)
 # AI RESPONSE FUNCTION
 # ===============================
 
-def generate_response(task, user_input, tone, chat_history):
+def generate_response(task, user_input, tone, history):
 
-    if chat_history is None:
-        chat_history = []
+    if history is None:
+        history = []
 
     prompt = f"""
     Task: {task}
@@ -32,9 +31,10 @@ def generate_response(task, user_input, tone, chat_history):
         {"role": "system", "content": "You are a professional AI business assistant."}
     ]
 
-    # Add previous messages properly
-    for message in chat_history:
-        messages.append(message)
+    # Add previous chat correctly
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
 
     messages.append({"role": "user", "content": prompt})
 
@@ -46,43 +46,41 @@ def generate_response(task, user_input, tone, chat_history):
 
     answer = response.choices[0].message.content
 
-    # Append properly formatted messages
-    chat_history.append({"role": "user", "content": user_input})
-    chat_history.append({"role": "assistant", "content": answer})
+    history.append((user_input, answer))
 
-    return chat_history, chat_history
+    return history, history
 
 
 # ===============================
 # PDF DOWNLOAD FUNCTION
 # ===============================
 
-def download_pdf(chat_history):
+def download_pdf(history):
 
     file_path = "AI_Agent_Chat.pdf"
     doc = SimpleDocTemplate(file_path)
     elements = []
     styles = getSampleStyleSheet()
 
-    for message in chat_history:
-        role = message["role"]
-        content = message["content"]
-        elements.append(Paragraph(f"<b>{role.capitalize()}:</b> {content}", styles["Normal"]))
+    for user_msg, bot_msg in history:
+        elements.append(Paragraph(f"<b>User:</b> {user_msg}", styles["Normal"]))
         elements.append(Spacer(1, 0.3 * inch))
+        elements.append(Paragraph(f"<b>AI:</b> {bot_msg}", styles["Normal"]))
+        elements.append(Spacer(1, 0.5 * inch))
 
     doc.build(elements)
     return file_path
 
 
 # ===============================
-# GRADIO UI
+# GRADIO UI (Compatible Version)
 # ===============================
 
 with gr.Blocks() as demo:
 
     gr.Markdown("# 💰 Freelancing AI Agent Pro")
 
-    chatbot = gr.Chatbot(type="messages")  # IMPORTANT FIX
+    chatbot = gr.Chatbot()  # ✅ NO type="messages"
     state = gr.State([])
 
     task = gr.Dropdown(
@@ -96,6 +94,7 @@ with gr.Blocks() as demo:
     )
 
     user_input = gr.Textbox(label="Enter Your Request")
+
     generate_btn = gr.Button("Generate")
     download_btn = gr.Button("Download as PDF")