Update app.py

app.py

@@ -1,29 +1,18 @@
+import uuid
+import os
 from groq import Groq
 import gradio as gr
-import os
-import speech_recognition as sr
-
-# Securely retrieve the Groq API key from Hugging Face Secrets
-groqkey = os.getenv("groqkey")
-
-# Ensure the API key exists
-if not groqkey:
-    raise ValueError("🚨 Error: 'groqkey' is missing! Make sure it is set in Hugging Face Secrets.")
+from gtts import gTTS
 
-# Initialize the Groq client
-client = Groq(api_key=groqkey)
-
-# Define chatbot response function
-def get_chatbot_response(user_message, insurance_type, country, input_type, image, voice, language, conversation_history):
-    """Fetch insurance-related responses from the AI."""
+client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
+def get_chatbot_response(user_message, country, language, conversation_history):
     system_message = (
-        f"You are an insurance expert providing accurate information on {insurance_type} insurance in {country}. "
-        f"Respond in {language}. Your responses should be clear and informative but must not be considered official legal or financial advice. "
-        "Always use factual information about policies, coverage, and common practices."
+        f"You are an insurance expert specializing in providing concise and accurate information about insurance policies, claims, and regulations based on the laws in {country}. "
+        f"Respond in {language}. Provide clear, factual information without offering personal advice or opinions. "
+        "Include relevant policy details, coverage options, or regulatory references when possible."
     )
 
-    # Maintain conversation history
     if conversation_history:
         if conversation_history[0]["role"] == "system":
             conversation_history[0]["content"] = system_message
@@ -31,22 +20,16 @@ def get_chatbot_response(user_message, insurance_type, country, input_type, imag
             conversation_history.insert(0, {"role": "system", "content": system_message})
     else:
         conversation_history.append({"role": "system", "content": system_message})
+    
+    conversation_history.append({"role": "user", "content": user_message})
 
-    # Handle input type
-    if input_type == "Text":
-        conversation_history.append({"role": "user", "content": user_message})
-    elif input_type == "Image" and image:
-        conversation_history.append({"role": "user", "content": "[User uploaded an image]"})
-    elif input_type == "Voice" and voice:
-        transcribed_text = transcribe_audio(voice)
-        conversation_history.append({"role": "user", "content": transcribed_text})
-
-    # API call to Groq
     completion = client.chat.completions.create(
-        model="mixtral-8x7b",
+        model="deepseek-r1-distill-llama-70b",
         messages=conversation_history,
         temperature=0.3,
-        top_p=0.95
+        top_p=0.95,
+        stream=True,
+        reasoning_format="hidden"
     )
 
     response = ""
@@ -56,134 +39,174 @@ def get_chatbot_response(user_message, insurance_type, country, input_type, imag
     conversation_history.append({"role": "assistant", "content": response})
 
     chat_display = [
-        (msg["content"], conversation_history[i + 1]["content"])
+        (msg["content"], conversation_history[i + 1]["content"]) 
         for i, msg in enumerate(conversation_history[:-1]) if msg["role"] == "user"
     ]
-
+    
     return conversation_history, chat_display
 
-# Convert voice input to text
-def transcribe_audio(audio):
-    recognizer = sr.Recognizer()
-    with sr.AudioFile(audio) as source:
-        audio_data = recognizer.record(source)
-        try:
-            text = recognizer.recognize_google(audio_data)
-            return text
-        except sr.UnknownValueError:
-            return "Sorry, I could not understand the audio."
-        except sr.RequestError:
-            return "Error processing the audio."
-
-# Styling and Theme
-theme = gr.themes.Base().set(
-    body_background_fill="linear-gradient(to right, #001F3F, #0052CC)",  # Dark navy to royal blue
-    button_primary_background_fill="linear-gradient(135deg, #FFD700, #FFAA00)",  # Golden gradient
-    button_primary_text_color="black"
+def text_to_audio(conversation_history, language):
+    lang_map = {
+        "English": "en",
+        "Spanish": "es",
+        "French": "fr",
+        "German": "de",
+        "Hindi": "hi",
+        "Mandarin": "zh-cn",
+        "Arabic": "ar"
+    }
+    lang_code = lang_map.get(language, "en")
+    
+    conversation_text = ""
+    for msg in conversation_history:
+        if msg["role"] == "user":
+            conversation_text += f"You said: {msg['content']}\n"
+        elif msg["role"] == "assistant":
+            conversation_text += f"AI Insurance Chatbot responded with: {msg['content']}\n"
+    
+    if not conversation_text.strip():
+        return None
+
+    tts = gTTS(text=conversation_text, lang=lang_code)
+    audio_filename = f"response_{uuid.uuid4().hex}.mp3"
+    tts.save(audio_filename)
+    return audio_filename
+
+
+theme = gr.themes.Ocean(
+    text_size="lg",
+    font=[gr.themes.GoogleFont('DM Sans'), 'ui-sans-serif', 'system-ui', 'sans-serif'],
+).set(
+    body_text_size='*text_md',
+    background_fill_secondary='*secondary_100',
+    chatbot_text_size='*text_md',
+    input_radius='*radius_md',
+    input_text_size='*text_md',
 )
 
 custom_css = """
-/* Title text with gold gradient */
 .title-text {
-    background: linear-gradient(90deg, #FFD700, #FFAA00);
+    background: #00A0B0;
     -webkit-background-clip: text;
     background-clip: text;
     color: transparent;
     -webkit-text-fill-color: transparent;
-    font-weight: 700;
-    font-size: 42px;
+    display: inline-block;
+    width: fit-content;
+    font-weight: bold;
     text-align: center;
-    text-transform: uppercase;
-    letter-spacing: 1px;
-    text-shadow: 1px 1px 5px rgba(255, 215, 0, 0.3);
+    font-size: 45px;
 }
-/* Body text - Gold on Dark Background */
-body, .gradio-container {
-    color: #FFD700 !important;
-    font-family: 'Inter', sans-serif;
+.insurance-button {
+    border: 1px solid #00A0B0;
+    background-color: transparent;
+    font-size: 15px;
+    padding: 5px 15px;
+    border-radius: 16px;
+    margin: 0 5px;
 }
-/* Chat messages - Minimalist design */
-.chat-message {
-    background: rgba(255, 255, 255, 0.05);
-    border-radius: 12px;
-    padding: 12px;
-    backdrop-filter: blur(8px);
+.insurance-button:hover {
+    background: linear-gradient(90deg, #00A0B0, #00FFEF) !important;
     color: white !important;
 }
-/* Buttons */
-button.primary {
-    background: linear-gradient(135deg, #FFD700, #FFAA00);
-    border: none;
-    color: black !important;
-    font-weight: bold;
-    text-transform: uppercase;
-    border-radius: 20px;
-    padding: 10px 20px;
-    transition: all 0.3s ease-in-out;
-}
-button.primary:hover {
-    transform: scale(1.1);
-    box-shadow: 0 0 15px rgba(255, 215, 0, 0.7);
+.country-language-container .gr-dropdown {
+    font-size: 10px;
+    max-height: 100px;
 }
 """
 
-# Build Gradio Interface
+def clear_history():
+    return []
+
+
 with gr.Blocks(theme=theme, css=custom_css) as demo:
+
     gr.HTML("<h2 class='title-text'>🛡️ AI Insurance Chatbot</h2>")
-    gr.Markdown("### Select your insurance type, country, and ask your question.")
+    gr.Markdown("### Welcome! Pick your country, choose a language, and describe your insurance-related query. We're here to assist you!")
 
-    with gr.Row():
-        insurance_type_input = gr.Dropdown(
-            ["Health", "Car", "Home", "Travel", "Life", "Disability", "Business", "Pet"],
-            label="Select Insurance Type",
-            interactive=True
-        )
+    with gr.Row(elem_classes="country-language-container"):
         country_input = gr.Dropdown(
-            ["USA", "Canada", "UK", "Germany", "France", "India", "Australia", "Other"],
-            label="Select Country",
+            ["Canada", "United States", "United Kingdom", "Spain", "France", "Germany", "India", "China", "Lebanon", "Other"],
+            label="🌍 Country for Insurance Regulations",
             interactive=True
         )
         language_input = gr.Dropdown(
-            ["English", "Spanish", "French"],
-            label="Select Language",
+            ["English", "Spanish", "French", "German", "Hindi", "Mandarin", "Arabic"],
+            label="🗣️ Language Output",
             interactive=True
         )
 
-    input_type = gr.Dropdown(
-        ["Text", "Image", "Voice"],
-        label="Select Input Type",
-        interactive=True
-    )
-
     custom_country_input = gr.Textbox(label="Enter Country (if not listed)", visible=False)
 
     conversation_state = gr.State([])
+    insurance_state = gr.State("")
+
+    chatbot = gr.Chatbot(label="💬 Chat History", min_height="500px")
+    chatbot.clear(fn=clear_history, outputs=conversation_state)
     
-    chatbot = gr.Chatbot(label="Chat History", type="messages")  # Fix: Set type to "messages"
+    with gr.Row():
+        auto_btn = gr.Button("🚗 Auto", elem_classes="insurance-button")
+        home_btn = gr.Button("🏠 Home", elem_classes="insurance-button")
+        health_btn = gr.Button("🏥 Health", elem_classes="insurance-button")
+        life_btn = gr.Button("❤️ Life", elem_classes="insurance-button")
+        travel_btn = gr.Button("✈️ Travel", elem_classes="insurance-button")
+        business_btn = gr.Button("🏢 Business", elem_classes="insurance-button")
+        liability_btn = gr.Button("⚖️ Liability", elem_classes="insurance-button")
+        pet_btn = gr.Button("🐾 Pet", elem_classes="insurance-button")
+
+    with gr.Row(equal_height=True):
+        scenario_input = gr.Textbox(
+            label="💡 Type your message...", 
+            placeholder="Describe your insurance query...", 
+            interactive=True,
+        )
+        submit_btn = gr.Button("Send", variant="primary", scale=0)
 
-    clear_button = gr.Button("Clear Chat History")
-    clear_button.click(lambda: [], outputs=[conversation_state, chatbot])
+    def update_insurance_selection(current, new_selection):
+        if "Insurance: " in current:
+            parts = current.split("Insurance: ", 1)
+            additional_text = parts[1] if len(parts) > 1 else ""
+        else:
+            additional_text = current
+        return f"{new_selection} Insurance: {additional_text}"
+
+    for btn, insurance in zip(
+        [auto_btn, home_btn, health_btn, life_btn, travel_btn, business_btn, liability_btn, pet_btn],
+        ["Auto", "Home", "Health", "Life", "Travel", "Business", "Liability", "Pet"]
+    ):
+        btn.click(lambda current, insurance=insurance: update_insurance_selection(current, insurance), inputs=scenario_input, outputs=scenario_input)
+
+    def submit(country, custom_country, language, scenario, conversation_state):
+        selected_country = custom_country if country == "Other" else country
+        updated_history, chat_display = get_chatbot_response(
+            scenario, selected_country, language, conversation_state
+        )
+        return updated_history, chat_display, ""
 
-    question_input = gr.Textbox(label="Ask your question...", placeholder="Describe your insurance question...", interactive=True, visible=False)
-    image_input = gr.File(label="Upload an image (optional)", type="filepath", visible=False)  # Fix: Change type to "filepath"
-    voice_input = gr.Audio(label="Speak your question (optional)", type="filepath", visible=False)
+    country_input.change(lambda c: gr.update(visible=c == "Other"), inputs=country_input, outputs=custom_country_input)
 
-    def toggle_input_visibility(input_type):
-        if input_type == "Text":
-            return [gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)]
-        elif input_type == "Image":
-            return [gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)]
-        elif input_type == "Voice":
-            return [gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)]
+    submit_btn.click(
+        submit, 
+        inputs=[country_input, custom_country_input, language_input, scenario_input, conversation_state], 
+        outputs=[conversation_state, chatbot, scenario_input]
+    )
 
-    input_type.change(toggle_input_visibility, inputs=[input_type], outputs=[question_input, image_input, voice_input])
+    scenario_input.submit(
+        fn=submit, 
+        inputs=[country_input, custom_country_input, language_input, scenario_input, conversation_state], 
+        outputs=[conversation_state, chatbot, scenario_input]
+    )
 
-    submit_btn = gr.Button("Send")
+    gr.HTML("<br><br>")
+    gr.Markdown("### Audio Output\nClick the **Read Conversation** button to have the entire conversation read aloud for you.")
 
-    submit_btn.click(
-        get_chatbot_response,
-        inputs=[question_input, insurance_type_input, country_input, input_type, image_input, voice_input, language_input, conversation_state],
-        outputs=[conversation_state, chatbot]
+    read_conversation_btn = gr.Button("🔊 Read Conversation", variant="primary")
+    response_audio_output = gr.Audio(label="Conversation Audio")
+    
+    read_conversation_btn.click(
+        fn=text_to_audio,
+        inputs=[conversation_state, language_input],
+        outputs=response_audio_output
     )
 
 demo.launch()