Spaces:
Sleeping
Sleeping
import gradio as gr | |
import google.generativeai as genai | |
# Configure Google Gemini API | |
genai.configure(api_key) # Replace with your actual API key | |
# Function to get a response from the Google Gemini model | |
def get_gemini_response(input_text): | |
# Check if the input is detailed enough | |
if len(input_text.split()) < 10: | |
return "Please provide a more detailed user story to help generate relevant needs and wants." | |
# Concise prompt to limit output to essential insights | |
input_prompt = f""" | |
Based on the user story "{input_text}", briefly extract any unarticulated needs and wants. | |
Only provide essential needs and wants directly relevant to the given story. Do not speculate or over-extrapolate. | |
Needs and Wants: | |
""" | |
# Generate the content based on text input | |
model = genai.GenerativeModel('gemini-1.5-flash') | |
response = model.generate_content([input_text, input_prompt]) | |
return response.text | |
# Gradio interface function | |
def extract_needs_and_wants(user_story): | |
try: | |
return get_gemini_response(user_story) | |
except Exception as e: | |
return f"Error: {str(e)}" | |
# Create the Gradio interface | |
import gradio as gr | |
interface = gr.Interface( | |
fn=extract_needs_and_wants, | |
inputs="text", | |
outputs="text", | |
title="Unarticulated Needs & Wants Extractor", | |
description="**Author:** VictorDaniel\n\nEnter a detailed user story to extract the unarticulated needs and wants.", | |
examples=[["The user often speaks about wanting to improve their health but is hesitant to join a gym."]] | |
) | |
interface.launch() | |
# Launch the Gradio app | |
interface.launch() |