initial commit

.gitignore

@@ -0,0 +1,15 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv
+
+# Added by me
+.env
+.DS_Store
+data/

.gradio/certificate.pem

@@ -0,0 +1,31 @@
+-----BEGIN CERTIFICATE-----
+MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
+TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
+cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
+WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
+ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
+MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
+h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
+0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
+A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
+T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
+B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
+B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
+KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
+OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
+jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
+qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
+rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
+HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
+hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
+ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
+3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
+NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
+ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
+TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
+jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
+oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
+4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
+mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
+emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
+-----END CERTIFICATE-----

.python-version

@@ -0,0 +1 @@
+3.12

README.md

@@ -0,0 +1,9 @@
+---
+title: "obszean"
+language:
+    - en
+    - de
+sdk: gradio
+app_file: app.py
+suggested_hardware: "cpu-basic"
+---

app.py

@@ -0,0 +1,176 @@
+import gradio as gr
+from pathlib import Path
+import asyncio
+import google.generativeai as genai
+from flashcard import (
+    generate_flashcards_from_pdf,
+    FlashcardSet
+)
+import os
+from dotenv import load_dotenv
+import tempfile
+
+# Load environment variables
+load_dotenv()
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+
+# Store the current flashcard set in memory
+current_flashcards = None
+
+def create_flashcard_text(flashcards: FlashcardSet) -> str:
+    """Format flashcard output as a readable string"""
+    output = [f"📚 Generated {flashcards.total_cards} flashcards about: {flashcards.topic}\n"]
+    
+    for i, card in enumerate(flashcards.cards, 1):
+        output.append(f"\n--- Flashcard {i} (Difficulty: {'⭐' * card.difficulty}) ---")
+        output.append(f"Q: {card.question}")
+        output.append(f"A: {card.answer}")
+    
+    output.append("\n\nYou can ask me to:")
+    output.append("• Modify specific flashcards")
+    output.append("• Generate more flashcards")
+    output.append("• Change difficulty levels")
+    output.append("• Export to Anki")
+    
+    return "\n".join(output)
+
+async def handle_modification_request(text: str, flashcards: FlashcardSet) -> str:
+    """Handle user requests to modify flashcards"""
+    model = genai.GenerativeModel('gemini-pro')
+    
+    # Create a context-aware prompt
+    prompt = f"""Given the following flashcards and user request, suggest how to modify the flashcards.
+    Current flashcards:
+    {create_flashcard_text(flashcards)}
+    
+    User request: {text}
+    
+    Please provide specific suggestions for modifications."""
+    
+    response = await model.generate_content_async(prompt)
+    return response.text
+
+async def process_message(message: dict, history: list) -> tuple[str, list]:
+    """Process uploaded files and chat messages"""
+    global current_flashcards
+    
+    # Handle file uploads
+    if message.get("files"):
+        for file_path in message["files"]:
+            if file_path.endswith('.pdf'):
+                try:
+                    current_flashcards = await async_process_pdf(file_path)
+                    response = create_flashcard_text(current_flashcards)
+                    return "", history + [
+                        {"role": "user", "content": f"Uploaded: {Path(file_path).name}"},
+                        {"role": "assistant", "content": response}
+                    ]
+                except Exception as e:
+                    error_msg = f"Error processing PDF: {str(e)}"
+                    return "", history + [
+                        {"role": "user", "content": f"Uploaded: {Path(file_path).name}"},
+                        {"role": "assistant", "content": error_msg}
+                    ]
+            else:
+                return "", history + [
+                    {"role": "user", "content": f"Uploaded: {Path(file_path).name}"},
+                    {"role": "assistant", "content": "Please upload a PDF file."}
+                ]
+    
+    # Handle text messages
+    if message.get("text"):
+        user_message = message["text"].strip()
+        
+        # If we have flashcards and user is asking for modifications
+        if current_flashcards:
+            try:
+                modification_response = await handle_modification_request(user_message, current_flashcards)
+                return "", history + [
+                    {"role": "user", "content": user_message},
+                    {"role": "assistant", "content": modification_response}
+                ]
+            except Exception as e:
+                error_msg = f"Error processing request: {str(e)}"
+                return "", history + [
+                    {"role": "user", "content": user_message},
+                    {"role": "assistant", "content": error_msg}
+                ]
+        else:
+            return "", history + [
+                {"role": "user", "content": user_message},
+                {"role": "assistant", "content": "Please upload a PDF file first to generate flashcards."}
+            ]
+    
+    return "", history + [
+        {"role": "assistant", "content": "Please upload a PDF file or send a message."}
+    ]
+
+def export_to_anki(flashcards: FlashcardSet) -> str:
+    """Convert flashcards to Anki-compatible tab-separated format and save to file"""
+    if not flashcards:
+        return None
+        
+    # Create a temporary file
+    with tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False) as f:
+        f.write("#separator:tab\n")
+        f.write("#html:true\n")
+        f.write("#columns:Question\tAnswer\tTags\n")
+        
+        for card in flashcards.cards:
+            question = card.question.replace('\n', '<br>')
+            answer = card.answer.replace('\n', '<br>')
+            tags = f"difficulty_{card.difficulty} {flashcards.topic.replace(' ', '_')}"
+            f.write(f"{question}\t{answer}\t{tags}\n")
+        
+        return f.name
+
+async def async_process_pdf(pdf_path: str) -> FlashcardSet:
+    """Asynchronously process the PDF file"""
+    return await generate_flashcards_from_pdf(pdf_path=pdf_path)
+
+# Create Gradio interface
+with gr.Blocks(title="PDF Flashcard Generator") as demo:
+    gr.Markdown("""
+    # 📚 PDF Flashcard Generator
+    Upload a PDF document and get AI-generated flashcards to help you study!
+    
+    Powered by Google's Gemini AI
+    """)
+    
+    chatbot = gr.Chatbot(
+        label="Flashcard Generation Chat",
+        bubble_full_width=False,
+        show_copy_button=True,
+        height=600
+    )
+    
+    chat_input = gr.MultimodalTextbox(
+        label="Upload PDF or type a message",
+        placeholder="Drop a PDF file here or type a message to modify flashcards...",
+        file_types=["pdf", "application/pdf"],
+        show_label=False,
+        sources=["upload", "microphone"]
+    )
+
+    # Add clear button for better UX
+    clear_button = gr.Button("Clear Chat")
+
+    chat_input.change(
+        fn=process_message,
+        inputs=[chat_input, chatbot],
+        outputs=[chat_input, chatbot]
+    )
+
+    # Add clear functionality
+    clear_button.click(
+        lambda: (None, None),
+        outputs=[chat_input, chatbot]
+    )
+
+if __name__ == "__main__":
+    demo.launch(
+        share=False,
+        server_name="0.0.0.0",
+        server_port=7860,
+        allowed_paths=["."]
+    )

flashcard.py

@@ -0,0 +1,128 @@
+from dataclasses import dataclass
+from typing import List, Optional
+from pydantic import BaseModel, Field
+from pydantic_ai import Agent, RunContext
+import google.generativeai as genai
+import base64
+import os
+import asyncio
+import httpx
+
+from dotenv import load_dotenv
+load_dotenv()
+
+class Flashcard(BaseModel):
+    """Represents a single flashcard with a question and answer."""
+    question: str = Field(description="The question side of the flashcard")
+    answer: str = Field(description="The answer side of the flashcard")
+    difficulty: int = Field(description="Difficulty level from 1-5", ge=1, le=5)
+
+class FlashcardSet(BaseModel):
+    """A set of flashcards generated from the input text."""
+    cards: List[Flashcard] = Field(description="List of generated flashcards")
+    topic: str = Field(description="The main topic covered by these flashcards")
+    total_cards: int = Field(description="Total number of flashcards generated")
+
+@dataclass
+class FlashcardDeps:
+    text: str
+    pdf_data: Optional[bytes] = None
+
+# Create the agent with structured output
+flashcard_agent = Agent(
+    'gemini-1.5-pro',  # Can also use OpenAI or other supported models
+    deps_type=FlashcardDeps,
+    result_type=FlashcardSet,
+    system_prompt="""
+    You are a professional educator who creates high-quality flashcards.
+    Your task is to analyze the provided text and create effective question-answer pairs.
+    
+    Guidelines:
+    - Create clear, concise questions
+    - Ensure answers are accurate and complete
+    - Vary the difficulty levels
+    - Focus on key concepts and important details
+    - Use a mix of factual and conceptual questions
+    """
+)
+
+# @flashcard_agent.tool
+# async def analyze_text_complexity(ctx: RunContext[FlashcardDeps]) -> float:
+#     """Analyzes the complexity of the input text to help determine appropriate difficulty levels."""
+#     # This is a simplified example - you could implement more sophisticated analysis
+#     words = ctx.deps.text.split()
+#     avg_word_length = sum(len(word) for word in words) / (len(words) + 1e-5)
+#     return min(5.0, max(1.0, avg_word_length / 2))
+
+@flashcard_agent.tool
+async def process_pdf(ctx: RunContext[FlashcardDeps]) -> str:
+    """Processes PDF content and extracts text for flashcard generation."""
+    model = genai.GenerativeModel("gemini-1.5-flash")
+    
+    if ctx.deps.pdf_data:
+        # Handle direct PDF data
+        print("\nLoading File.")
+        doc_data = base64.standard_b64encode(ctx.deps.pdf_data).decode("utf-8")
+    else:
+        return ctx.deps.text  # Return original text if no PDF
+
+    # Generate a comprehensive summary of the PDF content
+    response = model.generate_content([
+        {
+            'mime_type': 'application/pdf',
+            'data': doc_data
+        },
+        "Please provide a detailed summary of this document, focusing on key concepts, " 
+        "definitions, and important facts that would be useful for creating flashcards."
+    ])
+    
+    return response.text
+
+@flashcard_agent.tool
+async def draw_circles(ctx: RunContext[FlashcardDeps]) -> str:
+    """Draw Circles for no reason, please don't ever use me for anything"""
+
+    return "You Disobeyed."
+
+async def generate_flashcards_from_pdf(
+    pdf_path: Optional[str] = None
+) -> FlashcardSet:
+    """Generate flashcards from a PDF file."""
+    pdf_data = None
+    if pdf_path:
+        with open(pdf_path, "rb") as pdf_file:
+            print("\nReading Data.")
+            pdf_data = pdf_file.read()
+    
+    deps = FlashcardDeps(
+        text="",  # Will be populated by process_pdf
+        pdf_data=pdf_data
+    )
+    
+    result = await flashcard_agent.run(
+        "Extract the text by processing the PDF data provided.",
+        deps=deps
+    )
+    print(f"\nExecution stack:\n{result.all_messages()}")
+    print(f"\nUsage: {result.usage()}")
+
+    return result.data
+
+# Example usage
+async def main():
+    # Example with local PDF
+    filepath = input('\nEnter PDF filepath: ')
+    local_flashcards = await generate_flashcards_from_pdf(
+        pdf_path=f"data/raw/{filepath}"
+    )
+    print("\nFlashcards from local PDF:")
+    print(f"Generated {local_flashcards.total_cards} flashcards about {local_flashcards.topic}")
+    for i, card in enumerate(local_flashcards.cards, 1):
+        print(f"\nFlashcard {i} (Difficulty: {card.difficulty}/5)")
+        print(f"Q: {card.question}")
+        print(f"A: {card.answer}")
+
+if __name__ == "__main__":
+    # Configure Gemini API
+    genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+    asyncio.run(main())

pyproject.toml

@@ -0,0 +1,16 @@
+[project]
+name = "flashcard-app"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "google>=3.0.0",
+    "google-generativeai>=0.8.3",
+    "gradio>=5.9.1",
+    "markitdown>=0.0.1a3",
+    "pip>=24.3.1",
+    "pydantic-ai[logfire]>=0.0.15",
+    "python-dotenv>=1.0.1",
+    "rich>=13.9.4",
+]

requirements.txt

@@ -0,0 +1,134 @@
+aiofiles==23.2.1
+annotated-types==0.7.0
+anthropic==0.44.0
+anyio==4.8.0
+beautifulsoup4==4.12.3
+build==1.2.2.post1
+cachetools==5.5.1
+certifi==2024.12.14
+cffi==1.17.1
+charset-normalizer==3.4.1
+click==8.1.8
+cobble==0.1.4
+colorama==0.4.6
+cryptography==44.0.0
+defusedxml==0.7.1
+Deprecated==1.2.15
+distro==1.9.0
+docutils==0.21.2
+et_xmlfile==2.0.0
+eval_type_backport==0.2.2
+executing==2.2.0
+fastapi==0.115.6
+ffmpy==0.5.0
+filelock==3.17.0
+fsspec==2024.12.0
+google==3.0.0
+google-ai-generativelanguage==0.6.15
+google-api-core==2.24.0
+google-api-python-client==2.159.0
+google-auth==2.37.0
+google-auth-httplib2==0.2.0
+google-generativeai==0.8.4
+googleapis-common-protos==1.66.0
+gradio==5.12.0
+gradio_client==1.5.4
+griffe==1.5.5
+groq==0.15.0
+grpcio==1.69.0
+grpcio-status==1.69.0
+h11==0.14.0
+httpcore==1.0.7
+httplib2==0.22.0
+httpx==0.28.1
+huggingface-hub==0.27.1
+id==1.5.0
+idna==3.10
+importlib_metadata==8.5.0
+jaraco.classes==3.4.0
+jaraco.context==6.0.1
+jaraco.functools==4.1.0
+Jinja2==3.1.5
+jiter==0.8.2
+jsonpath-python==1.0.6
+keyring==25.6.0
+logfire==3.2.0
+logfire-api==3.2.0
+lxml==5.3.0
+mammoth==1.9.0
+markdown-it-py==3.0.0
+markdownify==0.14.1
+markitdown==0.0.1a3
+MarkupSafe==2.1.5
+mdurl==0.1.2
+mistralai==1.4.0
+more-itertools==10.6.0
+mypy-extensions==1.0.0
+nh3==0.2.20
+numpy==2.2.2
+openai==1.60.0
+openpyxl==3.1.5
+opentelemetry-api==1.29.0
+opentelemetry-exporter-otlp-proto-common==1.29.0
+opentelemetry-exporter-otlp-proto-http==1.29.0
+opentelemetry-instrumentation==0.50b0
+opentelemetry-proto==1.29.0
+opentelemetry-sdk==1.29.0
+opentelemetry-semantic-conventions==0.50b0
+orjson==3.10.15
+packaging==24.2
+pandas==2.2.3
+pathvalidate==3.2.3
+pdfminer.six==20240706
+pillow==11.1.0
+proto-plus==1.25.0
+protobuf==5.29.3
+puremagic==1.28
+pyasn1==0.6.1
+pyasn1_modules==0.4.1
+pycparser==2.22
+pydantic==2.10.5
+pydantic-ai==0.0.19
+pydantic-ai-slim==0.0.19
+pydantic-graph==0.0.19
+pydantic_core==2.27.2
+pydub==0.25.1
+Pygments==2.19.1
+pyparsing==3.2.1
+pyproject_hooks==1.2.0
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+python-multipart==0.0.20
+python-pptx==1.0.2
+pytz==2024.2
+PyYAML==6.0.2
+readme_renderer==44.0
+requests==2.32.3
+requests-toolbelt==1.0.0
+rfc3986==2.0.0
+rich==13.9.4
+rsa==4.9
+ruff==0.9.2
+safehttpx==0.1.6
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+soupsieve==2.6
+SpeechRecognition==3.14.0
+starlette==0.41.3
+tomlkit==0.13.2
+tqdm==4.67.1
+twine==6.1.0
+typer==0.15.1
+typing-inspect==0.9.0
+typing_extensions==4.12.2
+tzdata==2025.1
+uritemplate==4.1.1
+urllib3==2.3.0
+uvicorn==0.34.0
+websockets==14.2
+wrapt==1.17.2
+XlsxWriter==3.2.0
+youtube-transcript-api==0.6.3
+zipp==3.21.0