Update app.py

app.py

@@ -8,7 +8,25 @@ from google.generativeai.types import HarmBlockThreshold, HarmCategory
 # Configure Google API Key and model
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
 genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel("gemini-1.5-pro-002")
+MODEL_ID = "gemini-1.5-pro-002"
+model = GenerativeModel(MODEL_ID)
+
+example_model = GenerativeModel(
+    MODEL_ID,
+    system_instruction=[
+     "You are an advocate against gender-based violence.",
+    "Analyze the content for signs of gender discrimination and provide actionable advice."
+    ],
+)
+
+# Set model parameters
+generation_config = GenerationConfig(
+    temperature=0.9,
+    top_p=1.0,
+    top_k=32,
+    candidate_count=1,
+    max_output_tokens=8192,
+)
 
 # Safety and instruction settings
 safety_settings = {
@@ -17,24 +35,25 @@ safety_settings = {
     HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
     HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
 }
-system_instructions = [
-    "You are an advocate against gender-based violence.",
-    "Analyze the content for signs of gender discrimination and provide actionable advice."
-]
 
+prompt = """
+  Analyze this text for any instances of gender-based discrimination and provide tips
+"""
+
+# Set contents to send to the model
+contents = [prompt]
+
+
+# Prompt the model to generate content
+response = example_model.generate_content(
+    contents,
+    generation_config=generation_config,
+    safety_settings=safety_settings,
+)
 # Function to analyze text
 def analyze_text(text):
-    prompt = f"Analyze this text for any instances of gender-based discrimination and provide tips: {text}"
-    generation_config = genai.types.GenerationConfig(
-        temperature=0.5,
-        max_output_tokens=300,
-        stop_sequences=["\n"],
-        top_k=40,
-        top_p=0.9,
-        system_instructions=system_instructions,
-        safety_settings=safety_settings
-    )
-    response = model.generate_content([prompt], generation_config=generation_config)
+    prompt = f"Analyze this text for any instances of gender-based discrimination and provide tips{text}"
+    response 
     return response.text if response else "No response generated."
 
 # Function to analyze images
@@ -53,55 +72,19 @@ def analyze_image(image):
     response = model.generate_content([prompt, image], generation_config=generation_config)
     return response.text if response else "No response generated."
 
-# Function to analyze videos
-def analyze_video(video):
-    prompt = "Analyze this video for any instances of gender-based discrimination and provide tips."
-    clip = mp.VideoFileClip(video.name)
-    frame = clip.get_frame(1)  # Get a frame at t=1 second
-    image = Image.fromarray(frame)
-    image = image.convert("RGB")
-    generation_config = genai.types.GenerationConfig(
-        temperature=0.5,
-        max_output_tokens=300,
-        stop_sequences=["\n"],
-        top_k=40,
-        top_p=0.9,
-        system_instructions=system_instructions,
-        safety_settings=safety_settings
-    )
-    response = model.generate_content([prompt, image], generation_config=generation_config)
-    return response.text if response else "No response generated."
 
 # Gradio interface setup
 with gr.Blocks() as app:
     with gr.Tab("Text Analysis"):
         text_input = gr.Textbox(label="Enter Text", placeholder="Type here...", lines=4)
-        text_output = gr.Textbox(label="Analysis Output", lines=6)
         analyze_text_btn = gr.Button("Analyze Text")
+        text_output = gr.Textbox(label="Analysis Output", lines=6)
         analyze_text_btn.click(
             fn=analyze_text,
             inputs=text_input,
             outputs=text_output
         )
     
-    with gr.Tab("Image Analysis"):
-        image_input = gr.Image(label="Upload Image")
-        image_output = gr.Textbox(label="Analysis Output", lines=6)
-        analyze_image_btn = gr.Button("Analyze Image")
-        analyze_image_btn.click(
-            fn=analyze_image,
-            inputs=image_input,
-            outputs=image_output
-        )
-
-    with gr.Tab("Video Analysis"):
-        video_input = gr.Video(label="Upload Video")
-        video_output = gr.Textbox(label="Analysis Output", lines=6)
-        analyze_video_btn = gr.Button("Analyze Video")
-        analyze_video_btn.click(
-            fn=analyze_video,
-            inputs=video_input,
-            outputs=video_output
-        )
-
+ 
+ 
 app.launch()