Update app.py

app.py

@@ -1,54 +1,37 @@
-import os
 import gradio as gr
-import google.generativeai as genai
+import requests
 
-# Configure the Google Generative AI
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+# Define the function to interact with the API
+def generate_content(text):
+    url = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key=AIzaSyAflU5a8T3dCuIrkql737ZQtqWdrPmo0ts'
+    headers = {'Content-Type': 'application/json'}
+    data = {
+        "contents": [
+            {
+                "parts": [
+                    {
+                        "text": text
+                    }
+                ]
+            }
+        ]
+    }
+    
+    response = requests.post(url, headers=headers, json=data)
+    
+    if response.status_code == 200:
+        return response.json()  # You can customize this to return specific parts of the response
+    else:
+        return f"Error: {response.status_code} - {response.text}"
 
-# Create the model
-generation_config = {
-    "temperature": 1,
-    "top_p": 0.95,
-    "top_k": 64,
-    "max_output_tokens": 8192,
-    "response_mime_type": "text/plain",
-}
-
-model = genai.GenerativeModel(
-    model_name="gemini-1.5-flash",
-    generation_config=generation_config,
-)
-
-# Initialize a chat session
-chat_session = model.start_chat(
-    history=[
-        {
-            "role": "user",
-            "parts": [
-                "tell me a story about me",
-            ],
-        },
-        {
-            "role": "model",
-            "parts": [
-                "Please tell me a little bit about yourself! I need some information to write a story about you. For example:\n\n* **What are your hobbies and interests?** \n* **What are some of your favorite things?** \n* **What kind of story do you want to hear?** (Adventure, romance, mystery, etc.)\n* **Is there anything unique or interesting about you that you'd like to see in the story?**\n\nOnce I know more about you, I can write a fun and engaging story just for you! 😊 \n",
-            ],
-        },
-    ]
-)
-
-def generate_story(user_input):
-    response = chat_session.send_message(user_input)
-    return response.text
-
-# Create a Gradio Interface
+# Create a Gradio interface
 iface = gr.Interface(
-    fn=generate_story,
-    inputs=gr.Textbox(lines=4, placeholder="Enter some information about yourself..."),
-    outputs="text",
-    title="AI Story Generator",
-    description="Enter some details about yourself, and the AI will generate a personalized story for you!"
+    fn=generate_content,
+    inputs="text",
+    outputs="json",
+    title="AI Content Generator",
+    description="Enter a prompt to generate content using the Google Gemini API."
 )
 
-# Launch the interface
-iface.launch()
+# Launch the Gradio app
+iface.launch()