Spaces:

giseldo
/

study_plan_v2

Sleeping

App Files Files Community

giseldo commited on Mar 2, 2025

Commit

a5e365d

verified ·

1 Parent(s): 01ec90c

Update app.py

Browse files

Files changed (1) hide show

app.py +167 -17

app.py CHANGED Viewed

@@ -5,6 +5,58 @@ import markdown
 import tempfile
 import os
 import io
 groqllm = LLM(model="groq/llama-3.3-70b-versatile")
@@ -67,20 +119,118 @@ def executar_equipe_interface(disciplina, assunto, topicos_str):
     saida_final = f"{saida1}\n\n{saida2}\n\n{saida3}"
-    return saida_final
-iface = gr.Interface(
-    fn=executar_equipe_interface,
-    inputs=[
-        gr.Textbox(label="Disciplina", value="Matemática"),
-        gr.Textbox(label="Assunto", value="Funções"),
-        gr.Textbox(label="Tópicos (separados por vírgula)", value="Função quadrática, Função exponencial, Função logarítmica")
-    ],
-    outputs=[
-        gr.Markdown(label="Material Completo (Markdown)")
-    ],
-    title="Interface - Material de Estudos",
-    description="Interface para gerar material de estudo."
-)
-iface.launch()

 import tempfile
 import os
 import io
+import hashlib
+# Simple user database - In a production system, use a secure database
+USER_DATABASE = {
+    "admin": "8c6976e5b5410415bde908bd4dee15dfb167a9c873fc4bb8a81f6f2ab448a918",  # admin
+    "user": "04f8996da763b7a969b1028ee3007569eaf3a635486ddab211d512c85b9df8fb",    # user123
+    "test": "9f86d081884c7d659a2feaa0c55ad015a3bf4f1b2b0b822cd15d6c15b0f00a08"     # test
+}
+# Password hashing function
+def hash_password(password):
+    return hashlib.sha256(password.encode()).hexdigest()
+# Login function
+def login(username, password):
+    if username in USER_DATABASE and USER_DATABASE[username] == hash_password(password):
+        return True, f"Bem-vindo, {username}!"
+    else:
+        return False, "Credenciais inválidas. Tente novamente."
+# Function to create PDF from markdown
+def markdown_to_pdf(markdown_text):
+    """
+    Convert markdown text to PDF content
+    """
+    # Convert markdown to HTML
+    html_text = markdown.markdown(markdown_text)
+    # Add some basic styling
+    styled_html = f"""
+    <html>
+    <head>
+        <style>
+            body {{ font-family: Arial, sans-serif; margin: 20px; }}
+            h1 {{ color: #2c3e50; }}
+            h2 {{ color: #3498db; }}
+            h3 {{ color: #2980b9; }}
+            a {{ color: #3498db; }}
+        </style>
+    </head>
+    <body>
+        {html_text}
+    </body>
+    </html>
+    """
+    # Create PDF from HTML content
+    pdf_output = io.BytesIO()
+    pisa.CreatePDF(styled_html, dest=pdf_output)
+    pdf_output.seek(0)
+    return pdf_output
 groqllm = LLM(model="groq/llama-3.3-70b-versatile")
     saida_final = f"{saida1}\n\n{saida2}\n\n{saida3}"
+    # Generate PDF from the output
+    pdf_output = markdown_to_pdf(saida_final)
+    # Save the PDF to a temporary file
+    temp_dir = tempfile.gettempdir()
+    pdf_filename = f"material_estudo_{hash_password(saida_final)[:8]}.pdf"
+    pdf_path = os.path.join(temp_dir, pdf_filename)
+    with open(pdf_path, 'wb') as f:
+        f.write(pdf_output.getbuffer())
+    return saida_final, pdf_path
+# Main application function with both login and main interfaces
+def main_app():
+    with gr.Blocks(theme="soft") as app:
+        # Authentication state
+        is_logged_in = gr.State(False)
+        current_user = gr.State("")
+        # Login Interface
+        with gr.Group(visible=True) as login_block:
+            gr.Markdown("# Sistema de Material de Estudo")
+            gr.Markdown("Por favor, faça login para continuar")
+            username_input = gr.Textbox(label="Usuário")
+            password_input = gr.Textbox(label="Senha", type="password")
+            login_button = gr.Button("Login")
+            login_message = gr.Textbox(label="Mensagem", interactive=False)
+            # Login credentials info (for demo purposes)
+            gr.Markdown("""
+            ### Credenciais para teste:
+            - Usuário: admin, Senha: admin
+            - Usuário: user, Senha: user123
+            - Usuário: test, Senha: test
+            """)
+        # Main Application Interface
+        with gr.Group(visible=False) as main_block:
+            gr.Markdown("# Gerador de Material de Estudos")
+            user_info = gr.Markdown("Bem-vindo!")
+            logout_button = gr.Button("Logout")
+            with gr.Row():
+                disciplina_input = gr.Textbox(label="Disciplina", value="Matemática")
+                assunto_input = gr.Textbox(label="Assunto", value="Funções")
+                topicos_input = gr.Textbox(label="Tópicos (separados por vírgula)",
+                                           value="Função quadrática, Função exponencial, Função logarítmica")
+            generate_button = gr.Button("Gerar Material de Estudo")
+            with gr.Row():
+                markdown_output = gr.Markdown(label="Material Completo")
+                pdf_output = gr.File(label="Download PDF")
+        # Login function
+        def login_fn(username, password):
+            success, message = login(username, password)
+            if success:
+                return {
+                    login_block: gr.update(visible=False),
+                    main_block: gr.update(visible=True),
+                    login_message: message,
+                    user_info: f"### Usuário: {username}",
+                    is_logged_in: True,
+                    current_user: username
+                }
+            else:
+                return {
+                    login_message: message,
+                    is_logged_in: False
+                }
+        # Logout function
+        def logout_fn():
+            return {
+                login_block: gr.update(visible=True),
+                main_block: gr.update(visible=False),
+                username_input: "",
+                password_input: "",
+                login_message: "",
+                is_logged_in: False,
+                current_user: ""
+            }
+        # Generate material function
+        def generate_material(disciplina, assunto, topicos):
+            markdown, pdf = executar_equipe_interface(disciplina, assunto, topicos)
+            return markdown, pdf
+        # Connect buttons to functions
+        login_button.click(
+            login_fn,
+            inputs=[username_input, password_input],
+            outputs=[login_block, main_block, login_message, user_info, is_logged_in, current_user]
+        )
+        logout_button.click(
+            logout_fn,
+            outputs=[login_block, main_block, username_input, password_input,
+                     login_message, is_logged_in, current_user]
+        )
+        generate_button.click(
+            generate_material,
+            inputs=[disciplina_input, assunto_input, topicos_input],
+            outputs=[markdown_output, pdf_output]
+        )
+    return app
+# Launch the application
+if __name__ == "__main__":
+    app = main_app()
+    app.launch()