Update app.py

app.py

@@ -1,82 +1,48 @@
-import gradio as gr
-import google.generativeai as genai
-import PyPDF2
 import os
+import google.generativeai as genai
+import gradio as gr
 from dotenv import load_dotenv
 
-# API anahtarınızı yükleyin
-load_dotenv(override=True)
-api_key = os.getenv("GEMINI_API_KEY")
-genai.configure(api_key=api_key)
+# .env dosyasını yükleyin
+load_dotenv(dotenv_path='.env', override=True)
 
-# Gemini modelini başlat
-model = genai.GenerativeModel(model_name="gemini-1.5-flash",
-                              system_instruction="Sen bir belge sohbet asistanısın. Sorulara sana verilen belgeler içindeki bilgileri tarayarak yanıt vermelisin. Aksi belirtilmediği sürece yanıtları Türkçe olarak ver."
-                              )
+# API anahtarını ayarlayın
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
-def extract_text_from_pdf(pdf_files):
-    extracted_texts = []
-    for pdf_file in pdf_files:
-        with open(pdf_file.name, 'rb') as f:
-            pdf_reader = PyPDF2.PdfReader(f)
-            extracted_text = ""
-            for page in pdf_reader.pages:
-                text = page.extract_text()
-                if text:
-                    extracted_text += text
-            extracted_texts.append(extracted_text)
-    return extracted_texts
+# System instruction tanımlama
+system_instruction = """Sen Türk mitolojisini çok iyi bilen bir yapay zeka asistanısın ve adın bundan sonra Kam. Sana bu konuda soru soranlara bu konu hakkında derinlemesine bilgi vermeni istiyorum. Bunun dışındaki konulara girmemek."""
 
-def chat_with_documents(history, pdf_files):
-    if not history:
-        return history
-    
-    extracted_texts = extract_text_from_pdf(pdf_files)
-    user_input = history[-1]['content']
-    prompt = [user_input] + extracted_texts
-    response = model.generate_content(prompt)
-    
-    history.append({
-        'role': 'assistant', 
-        'content': response.text
-    })
-    return history
+def generate_text(prompt, model_name):
+    if not model_name:
+        return "Modeli Kontrol Ediniz"
+    try:
+        # Sistem talimatını prompt'a ekleyin
+        full_prompt = f"{system_instruction}\n\n{prompt}"
+        model = genai.GenerativeModel(model_name=model_name)
+        response = model.generate_content(full_prompt)
+        return response.text
+    except Exception as e:
+        return f"Hata Oluştu. Hata Kodu: {str(e)}"
 
+# gradio arayüzü
 with gr.Blocks() as demo:
-    gr.Markdown("# Belgelerinizle Sohbet Edin (Gemini API kullanarak)")
-    with gr.Accordion("Belgeler", open=False):
-        docs = gr.File(
-            label="Belgeleri Yükleyin",
-            file_count="multiple",
-            type="filepath"
+    gr.Markdown("# Geminichat")
+    gr.Markdown('Gemini\'ye sorular sorun')
+
+    with gr.Accordion("Modeller", open=False):
+        select_model = gr.Dropdown(
+            ['gemini-1.5-flash', 'gemini-exp-1114', 'gemini-1.5-pro'],
+            label="Modeller",
+            value="gemini-1.5-pro"
         )
-    chatbot = gr.Chatbot(type="messages")
-    msg = gr.Textbox(label="Mesajınız")
-    clear = gr.Button("Temizle")
-
-    def on_user_message(user_message, history):
-        history = history or []
-        history.append({
-            'role': 'user', 
-            'content': user_message
-        })
-        return "", history
-
-    def on_generate_response(history, docs_value):
-        if not docs_value:
-            history.append({
-                'role': 'assistant', 
-                'content': "Lütfen önce belge(leri) yükleyin."
-            })
-            return history
-        
-        if history:
-            history = chat_with_documents(history, docs_value)
-        return history
-
-    msg.submit(on_user_message, [msg, chatbot], [msg, chatbot]).then(
-        on_generate_response, [chatbot, docs], chatbot
-    )
-    clear.click(lambda: None, None, chatbot, queue=False)
+    with gr.Row():
+        text_input = gr.Textbox(label="Sorunuzu yazın.", placeholder="Enter tuşuna basarak gönderebilirsiniz")
+        with gr.Row():
+            btn = gr.Button("Sor")
+
+    with gr.Row():
+        text_output = gr.Textbox(label="Yanıtlar:")
+        btn.click(fn=generate_text, inputs=[text_input, select_model], outputs=text_output)
+        text_input.submit(fn=generate_text, inputs=[text_input, select_model], outputs=text_output)
     
-demo.launch()
+    demo.launch()