validation is added to deal with insufficient data

app.py

@@ -1,47 +1,49 @@
-# app.py
-
-import gradio as gr
-import google.generativeai as genai
-import os
-
-# Configure Google Gemini API
-genai.configure(api_key="AIzaSyDBGF5y7WqDB0SO7-UO6yjshiEZN3Hpt3g")  # Replace with your API key
-
-# Function to get a response from the Google Gemini model
-def get_gemini_response(input_text):
-    model = genai.GenerativeModel('gemini-1.5-flash')
-
-    # Input prompt for extracting unarticulated needs and wants
-    input_prompt = f"""
-    From the following user story, extract the unarticulated needs and wants.
-    User story: {input_text}
-    
-    Needs are the unspoken requirements or desires the person might not have expressed directly. 
-    Wants are the things the person wishes for but didn't explicitly say.
-    
-    Needs and Wants:
-    """
-
-    # Generate the content based on text input
-    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
-interface = gr.Interface(
-    fn=extract_needs_and_wants,
-    inputs="text",
-    outputs="text",
-    title="Unarticulated Needs & Wants Extractor",
-    description="Enter a user story to extract the unarticulated needs and wants using the Gemini model.",
-    examples=[["The user often speaks about wanting to improve their health but is hesitant to join a gym."]]
-)
-
-# Launch the Gradio app
-interface.launch()
+import gradio as gr
+import google.generativeai as genai
+
+# Configure Google Gemini API
+genai.configure(api_key="AIzaSyDBGF5y7WqDB0SO7-UO6yjshiEZN3Hpt3g")  # Replace with your API key
+
+# Function to get a response from the Google Gemini model
+def get_gemini_response(input_text):
+    model = genai.GenerativeModel('gemini-1.5-flash')
+
+    # Input prompt for extracting unarticulated needs and wants
+    input_prompt = f"""
+    From the following user story, extract the unarticulated needs and wants.
+    User story: {input_text}
+    
+    Needs are the unspoken requirements or desires the person might not have expressed directly. 
+    Wants are the things the person wishes for but didn't explicitly say.
+    
+    Needs and Wants:
+    """
+
+    # Generate the content based on text input
+    response = model.generate_content([input_text, input_prompt])
+    return response.text
+
+# Enhanced Gradio interface function with input validation
+def extract_needs_and_wants(user_story):
+    # Check if the input story is adequate (e.g., at least 20 characters or more than a few words)
+    if len(user_story.strip()) < 20 or len(user_story.split()) < 5:
+        return "Please provide a detailed user story with sufficient content for analysis."
+    
+    try:
+        # Process the input if it meets the criteria
+        return get_gemini_response(user_story)
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Create the Gradio interface
+interface = gr.Interface(
+    fn=extract_needs_and_wants,
+    inputs="text",
+    outputs="text",
+    title="Unarticulated Needs & Wants Extractor",
+    description="Enter a detailed user story to extract the unarticulated needs and wants using the Gemini model.",
+    examples=[["The user often speaks about wanting to improve their health but is hesitant to join a gym."]]
+)
+
+# Launch the Gradio app
+interface.launch()