Update app.py

app.py

@@ -1,35 +1,25 @@
-import openai
 import gradio as gr
+from transformers import pipeline
 
-# Set up OpenAI API key
-openai.api_key = 'your_openai_api_key_here'  # Replace with your OpenAI API key
+# Load your model from Hugging Face (for text generation)
+model = pipeline('text-generation', model='gpt2')  # Replace with the model you want
 
-# Function to generate code based on user input
+# Function to generate code (or text) based on user input
 def generate_code(input_text):
     try:
-        # Use GPT-3 to generate code
-        response = openai.Completion.create(
-            engine="text-davinci-003",
-            prompt=f"Generate a code based on the following request: {input_text}",
-            max_tokens=150,
-            n=1,
-            stop=None,
-            temperature=0.7
-        )
-        # Extract the response text
-        code = response.choices[0].text.strip()
-        return code
+        # Generate the output text based on input
+        output = model(input_text, max_length=150, num_return_sequences=1)[0]['generated_text']
+        return output
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Gradio Interface setup
+# Gradio interface
 iface = gr.Interface(
     fn=generate_code,  # Function to call
-    inputs="text",  # User input type (text input box)
-    outputs="text",  # Output type (text box for the generated code)
-    title="HandyAI - Your All Purpose AI",  # Title of the app
-    description="HandyAI generates code based on your request. Simply describe what you want, and it will generate code for you.",  # Description
-    theme="compact",  # Interface theme (compact, default, etc.)
+    inputs="text",  # Text input field for user to enter requests
+    outputs="text",  # Text output for generated code or text
+    title="HandyAI - Your All Purpose AI",
+    description="HandyAI can generate code or text based on your input request."
 )
 
 # Launch the interface