Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+import gradio as gr
 from google import genai
 import os
 
@@ -15,22 +15,18 @@ def generate_content(prompt):
         model="gemini-2.0-flash",
         contents=[prompt]
     )
-    return response.text
+    
+    # Format the result for better readability
+    result_text = response.text.strip()  # Remove extra spaces and line breaks
+    formatted_response = f"**AI Generated Content:**\n\n{result_text}"
+    return formatted_response
 
-# Streamlit app layout
-st.title("Gemini AI Content Generator")
-st.markdown("Provide a prompt and get an explanation from Gemini AI.")
-
-# Input from user
-prompt = st.text_area("Enter your prompt:", "How does AI work?")
-
-# Button to trigger content generation
-if st.button("Generate Content"):
-    if prompt:
-        # Get the response from the model
-        result = generate_content(prompt)
-        st.subheader("AI Response:")
-        st.markdown(result)
-    else:
-        st.warning("Please enter a prompt.")
+# Gradio interface
+interface = gr.Interface(fn=generate_content, 
+                         inputs="text", 
+                         outputs="markdown",  # Output as markdown for better formatting
+                         title="Gemini AI Content Generator",
+                         description="Provide a prompt and get an explanation from Gemini AI.\n\nThe AI will return a well-formatted response.")
 
+# Launch Gradio interface
+interface.launch()