Update app.py

app.py

@@ -1,16 +1,145 @@
 import os
 import gradio as gr
-from langchain.chat_models import ChatOpenAI
+from langchain_groq import ChatGroq
 from langchain import LLMChain, PromptTemplate
 from langchain.memory import ConversationBufferMemory
 
-OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
+GROQ_API_KEY=os.getenv('GROQ_API_KEY')
 
 template = """You are a helpful assistant to answer all user queries.
 {chat_history}
 User: {user_message}
 Chatbot:"""
 
+import os
+from groq import Groq
+import gradio as gr
+import logging
+
+# Set up logging configuration
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
+# Initialize the Groq client with API key from environment variables
+api_key = os.environ.get("GROQ_API_KEY")
+if not api_key:
+    logger.error("GROQ_API_KEY environment variable is not set.")
+    raise ValueError("GROQ_API_KEY environment variable is required.")
+client = Groq(api_key=api_key)
+
+# Default model name, can be overridden by environment variable
+MODEL_NAME = os.environ.get("MODEL_NAME", "llama3-8b-8192")
+
+def get_completion(user_input):
+    """
+    Generate a chat completion response using the Groq client.
+
+    Args:
+        user_input (str): The user's input query.
+
+    Returns:
+        str: The generated response or an error message.
+    """
+    try:
+        completion = client.chat.completions.create(
+            model=MODEL_NAME,
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": user_input}
+            ],
+            temperature=1,
+            max_tokens=1024,
+            top_p=1,
+            stream=False,  # Gradio does not support stream=True
+        )
+
+        # Extract the response
+        response = completion.choices[0].message.content.strip()
+        return response
+    except Exception as e:
+        logger.error(f"Error during completion: {e}")
+        return "Sorry, I encountered an error while processing your request."
+
+def launch_interface():
+    """
+    Launch the Gradio interface for the chatbot.
+    """
+    demo = gr.Interface(
+        fn=get_completion,
+        inputs=gr.Textbox(
+            label="Enter your query:",
+            placeholder="Ask me anything...",
+            lines=2,
+            max_lines=5
+        ),
+        outputs=gr.Textbox(
+            label="Response:",
+            lines=6,
+            max_lines=10
+        ),
+        title="Mr AI",
+        description="""
+        <style>
+        .gr-box {
+            border-radius: 10px;
+            border: 2px solid #007BFF;
+            padding: 15px;
+            background-color: #F8F9FA;
+        }
+        .gr-input {
+            font-size: 1.2em;
+            padding: 10px;
+            border-radius: 5px;
+            border: 2px solid #007BFF;
+            margin-bottom: 10px;
+        }
+        .gr-output {
+            font-size: 1.2em;
+            padding: 10px;
+            border-radius: 5px;
+            border: 2px solid #28A745;
+            background-color: #E9F7EF;
+        }
+        .gr-interface-title {
+            font-size: 2em;
+            font-weight: bold;
+            color: #007BFF;
+            text-align: center;
+            margin-bottom: 20px;
+        }
+        .gr-interface-description {
+            font-size: 1.2em;
+            color: #6C757D;
+            text-align: center;
+            margin-bottom: 20px;
+        }
+        .gr-button {
+            background-color: #007BFF;
+            color: white;
+            border-radius: 5px;
+            padding: 10px 20px;
+            font-size: 1em;
+            border: none;
+            cursor: pointer;
+            margin-top: 10px;
+        }
+        .gr-button:hover {
+            background-color: #0056b3;
+        }
+        </style>
+        <div class="gr-interface-title">Welcome to Mr AI</div>
+        <div class="gr-interface-description">Ask anything and get a helpful response.</div>
+        """,
+        allow_flagging="never",
+        live=True
+    )
+
+    logger.info("Starting Gradio interface")
+    demo.launch(share=True)
+
+if __name__ == "__main__":
+    launch_interface()
+
 prompt = PromptTemplate(
     input_variables=["chat_history", "user_message"], template=template
 )