| import streamlit as st | |
| import google.generativeai as genai | |
| from PIL import Image | |
| import io | |
| import base64 | |
| import pandas as pd | |
| import zipfile | |
| import os | |
| import PyPDF2 | |
| def encode_image(image): | |
| """Convert PIL Image to base64 string""" | |
| buffered = io.BytesIO() | |
| image.save(buffered, format="JPEG") | |
| image_bytes = buffered.getvalue() | |
| encoded_image = base64.b64encode(image_bytes).decode('utf-8') | |
| return encoded_image | |
| def read_file(uploaded_file): | |
| """Reads different file formats and returns content as string""" | |
| file_type = uploaded_file.name.split('.')[-1].lower() | |
| if file_type in ["txt", "html", "css", "php", "js", "py", "java", "c", "cpp"]: | |
| return uploaded_file.read().decode("utf-8") | |
| elif file_type in ["csv", "xlsx"]: | |
| df = pd.read_csv(uploaded_file) if file_type == "csv" else pd.read_excel(uploaded_file) | |
| return df.to_string() | |
| elif file_type == "pdf": | |
| reader = PyPDF2.PdfReader(uploaded_file) | |
| text = "".join([page.extract_text() for page in reader.pages if page.extract_text()]) | |
| return text | |
| elif file_type == "zip": | |
| with zipfile.ZipFile(uploaded_file, 'r') as zip_ref: | |
| file_list = zip_ref.namelist() | |
| extracted_text = "\n".join(file_list) | |
| return f"ZIP Inhalt: {extracted_text}" | |
| return "Nicht unterstütztes Dateiformat." | |
| st.set_page_config(page_title="Gemini AI Chat", layout="wide") | |
| st.title("🤖 Gemini AI Chat Interface") | |
| api_key = st.text_input("Enter Google AI API Key", type="password") | |
| model = st.selectbox("Select Model", [ | |
| "gemini-1.5-flash", "gemini-1.5-pro", "gemini-1.5-flash-8B", "gemini-1.5-pro-vision-latest", | |
| "gemini-1.0-pro", "gemini-1.0-pro-vision-latest", "gemini-2.0-pro-exp-02-05", "gemini-2.0-flash-lite", | |
| "gemini-2.0-flash-exp-image-generation", "gemini-2.0-flash", "gemini-2.0-flash-thinking-exp-01-21" | |
| ]) | |
| uploaded_file = st.file_uploader("Upload a file (optional)", type=["txt", "csv", "xlsx", "pdf", "zip", "html", "css", "php", "js", "py", "java", "c", "cpp"]) | |
| file_content = None | |
| if uploaded_file: | |
| file_content = read_file(uploaded_file) | |
| st.text_area("File Content Preview", file_content, height=200) | |
| user_input = st.text_area("Your message") | |
| if st.button("Send") and api_key: | |
| try: | |
| genai.configure(api_key=api_key) | |
| model_instance = genai.GenerativeModel(model_name=model) | |
| content = [{"text": user_input}] | |
| if file_content: | |
| content.append({"text": file_content}) | |
| response = model_instance.generate_content(content) | |
| st.markdown(response.text) | |
| except Exception as e: | |
| st.error(f"Error: {str(e)}") | |