removed redundant code

app.py

@@ -5,7 +5,6 @@ from langchain.memory import ConversationBufferMemory
 from langchain.chains import ConversationChain
 from langchain.prompts import PromptTemplate
 
-# Constants
 MODEL_NAME = "llama-3.3-70b-versatile"
 DEFAULT_API_KEY = os.getenv("GROQ_API_KEY", "")
 
@@ -34,18 +33,16 @@ AI Assistant:"""
         template=template
     )
     
-    conversation = ConversationChain(
+    return ConversationChain(
         llm=llm,
         memory=memory,
         prompt=prompt,
         verbose=False
     )
-    
-    return conversation
 
 conversation_chain = None
 
-def chat_function(message, history, api_key):
+def chat_function(message, api_key):
     global conversation_chain
     
     if not api_key:
@@ -58,37 +55,29 @@ def chat_function(message, history, api_key):
             return f"Error initializing chatbot: {str(e)}"
     
     try:
-        response = conversation_chain.predict(input=message)
-        return response
+        return conversation_chain.predict(input=message)
     except Exception as e:
         return f"Error: {str(e)}"
 
 def reset_conversation():
     global conversation_chain
     conversation_chain = None
-    return None
 
 with gr.Blocks(title="LLM based Chatbot") as demo:
     gr.Markdown("# 🤖 LLM based Chatbot")
     gr.Markdown("Chat with an AI assistant powered by LangChain and Groq")
     gr.Markdown(f"**Model:** `{MODEL_NAME}`")
     
-    # Only show API key input if not set in environment
     if not DEFAULT_API_KEY:
-        with gr.Row():
-            api_key_input = gr.Textbox(
-                label="Groq API Key",
-                placeholder="Enter your Groq API key here...",
-                type="password",
-                value=""
-            )
+        api_key_input = gr.Textbox(
+            label="Groq API Key",
+            placeholder="Enter your Groq API key here...",
+            type="password"
+        )
     else:
-        # Use hidden component with default key from environment
         api_key_input = gr.Textbox(
-            label="Groq API Key (Loaded from Environment)",
             type="password",
             value=DEFAULT_API_KEY,
-            interactive=False,
             visible=False
         )
     
@@ -102,20 +91,14 @@ with gr.Blocks(title="LLM based Chatbot") as demo:
         )
         submit_btn = gr.Button("Send", scale=1)
     
-    with gr.Row():
-        clear_btn = gr.Button("Clear Conversation")
+    clear_btn = gr.Button("Clear Conversation")
     
     def respond(message, chat_history, api_key):
         if not message.strip():
             return chat_history, ""
         
-        # Add user message
         chat_history.append({"role": "user", "content": message})
-        
-        # Get bot response
-        bot_message = chat_function(message, chat_history, api_key)
-        
-        # Add assistant message
+        bot_message = chat_function(message, api_key)
         chat_history.append({"role": "assistant", "content": bot_message})
         
         return chat_history, ""