Update app.py

app.py

@@ -1,21 +1,26 @@
-
-
-from transformers import pipeline, Conversation
 import gradio as gr
 
-# Load chatbot pipeline
-chatbot = pipeline("conversational", model="microsoft/DialoGPT-small")
-
-# Chat function for Gradio
-def chat_fn(message, history):
-    conversation = Conversation(message)
-    response = chatbot(conversation)
-    return response.generated_responses[-1]
+def calculator(num1, num2, operation):
+    if operation == "Addition (+)":
+        return num1 + num2
+    elif operation == "Subtraction (-)":
+        return num1 - num2
+    elif operation == "Multiplication (*)":
+        return num1 * num2
+    elif operation == "Division (/)" and num2 != 0:
+        return num1 / num2
+    else:
+        return "Error: Division by zero"
 
-# Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 Simple Chatbot (DialoGPT)")
-    chatbot_ui = gr.ChatInterface(fn=chat_fn)
+demo = gr.Interface(
+    fn=calculator,
+    inputs=[
+        gr.Number(label="Number 1"),
+        gr.Number(label="Number 2"),
+        gr.Radio(["Addition (+)", "Subtraction (-)", "Multiplication (*)", "Division (/)"], label="Operation")
+    ],
+    outputs="text",
+    title="Simple Calculator"
+)
 
-# Launch app
 demo.launch()