Spaces:

sarun25
/

test

Runtime error

test / app.py

Update app.py

ba0742d verified 6 months ago

1.34 kB

import gradio as gr
from google import genai
import os

# 1. Fetch the API Key from the environment (configured in Settings -> Secrets)
GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")

def generate_response(prompt):
    # Check if the API Key is provided
    if not GOOGLE_API_KEY:
        return "⚠️ Error: API Key not found! Please configure GOOGLE_API_KEY in the Settings -> Variables and secrets page of your Space."
    
    try:
        # 2. Initialise the Client using the new SDK
        client = genai.Client(api_key=GOOGLE_API_KEY)
        
        # 3. Send the prompt to Google AI Studio for processing
        response = client.models.generate_content(
            model='gemma-4-31b-it',
            contents=prompt
        )
        return response.text
    except Exception as e:
        return f"❌ An error occurred: {str(e)}"

# 4. Build the Gradio UI (Error fixed by removing allow_flagging)
demo = gr.Interface(
    fn=generate_response,
    inputs=gr.Textbox(lines=4, placeholder="Type your prompt here...", label="Prompt"),
    outputs=gr.Textbox(label="Response from Model"),
    title="⚡ My Gemini/Gemma App (Hosted on HF)",
    description="This application is hosted for free on Hugging Face Spaces, powered by the Google API using the gemma-4-31b-it model."
)

# Launch the web application
demo.launch()

	import gradio as gr
	from google import genai
	import os

	# 1. Fetch the API Key from the environment (configured in Settings -> Secrets)
	GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")

	def generate_response(prompt):
	# Check if the API Key is provided
	if not GOOGLE_API_KEY:
	return "⚠️ Error: API Key not found! Please configure GOOGLE_API_KEY in the Settings -> Variables and secrets page of your Space."

	try:
	# 2. Initialise the Client using the new SDK
	client = genai.Client(api_key=GOOGLE_API_KEY)

	# 3. Send the prompt to Google AI Studio for processing
	response = client.models.generate_content(
	model='gemma-4-31b-it',
	contents=prompt
	)
	return response.text
	except Exception as e:
	return f"❌ An error occurred: {str(e)}"

	# 4. Build the Gradio UI (Error fixed by removing allow_flagging)
	demo = gr.Interface(
	fn=generate_response,
	inputs=gr.Textbox(lines=4, placeholder="Type your prompt here...", label="Prompt"),
	outputs=gr.Textbox(label="Response from Model"),
	title="⚡ My Gemini/Gemma App (Hosted on HF)",
	description="This application is hosted for free on Hugging Face Spaces, powered by the Google API using the gemma-4-31b-it model."
	)

	# Launch the web application
	demo.launch()