File size: 2,811 Bytes
d62565a a13a171 d62565a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
import gradio as gr
import google.generativeai as genai
import os
# Set Google API Key
GOOGLE_API_KEY = "AIzaSyC9ySha6dLDOZirT3jaUOQC-cQOVCd7N1U"
genai.configure(api_key=GOOGLE_API_KEY)
# Load the Gemini Model
model = genai.GenerativeModel(model_name="models/gemini-2.0-flash")
def analyze_input(text, file):
try:
if file is not None:
with open(file, "r", encoding="utf-8") as f:
text = f.read()
elif not text.strip():
return "Error: Please enter text or upload a file.", ""
text = text[:2000] # Limit input text size
prompt = f"Analyze and summarize this document:\n\n{text}"
response = model.generate_content([prompt], stream=True) # Fix applied
# Collect streamed response
result = "".join([chunk.text for chunk in response])
word_count = len(text.split())
return result, f"Word Count: {word_count}"
except Exception as e:
return f"Error: {str(e)}", ""
# Function to clear inputs and outputs
def clear_inputs():
return "", None, "", "", None
# Function to generate downloadable text file
def generate_downloadable_file(text):
if text.strip():
file_path = "analysis_result.txt"
with open(file_path, "w", encoding="utf-8") as f:
f.write(text)
return file_path
else:
return None
# Create Gradio UI
with gr.Blocks(theme=gr.themes.Default()) as demo:
gr.Markdown("""
# **AI-Powered Text & File Analyzer**
Upload a `.txt` file or enter text manually to get an AI-generated analysis and summary.
""")
with gr.Row():
text_input = gr.Textbox(label="Enter Text", placeholder="Type or paste your text here...", lines=6)
file_input = gr.File(label="Upload Text File (.txt)", type="filepath")
output_text = gr.Textbox(label="Analysis Result", lines=10, interactive=False)
word_count_display = gr.Textbox(label="Word Count", interactive=False)
with gr.Row():
analyze_button = gr.Button("Analyze", variant="primary")
clear_button = gr.Button("Clear", variant="secondary")
with gr.Column():
gr.Markdown("Download Analysis Result")
with gr.Row():
download_button = gr.Button("Download Result", variant="success", size="sm")
download_file = gr.File(label="Click to Download", interactive=False)
# Button functionalities
analyze_button.click(analyze_input, inputs=[text_input, file_input], outputs=[output_text, word_count_display])
clear_button.click(clear_inputs, inputs=[], outputs=[text_input, file_input, output_text, word_count_display, download_file])
download_button.click(generate_downloadable_file, inputs=output_text, outputs=download_file)
# Launch the Gradio app
demo.launch() |