Update app.py

app.py

@@ -10,9 +10,11 @@ from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_excep
 import threading
 import os
 from io import BytesIO
+from docx import Document
+import markdown
 
 # Hugging Face variables
-GITHUB_TOKEN = os.environ.get('GITHUB_TOKEN')
+GIT_TOKEN = os.environ.get('GIT_TOKEN')
 GEMINI_API_KEY = os.environ.get('GEMINI_API_KEY')
 
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
@@ -20,7 +22,7 @@ app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 # Global variable to store generated file
 generated_file = None
 
-def fetch_git_files(git_url, personal_access_token, git_provider):
+def fetch_git_files(git_url, git_provider):
     try:
         # Parse the Git URL
         parts = git_url.split('/')
@@ -38,7 +40,7 @@ def fetch_git_files(git_url, personal_access_token, git_provider):
 
         # Set up headers with the personal access token
         headers = {
-            "Authorization": f"token {personal_access_token}",
+            "Authorization": f"token {GIT_TOKEN}",
             "Accept": "application/vnd.github.v3+json"
         }
 
@@ -120,34 +122,34 @@ def process_with_gemini(file_content, gemini_api_key):
     combined_result = "\n\n".join(results)
     return combined_result
 
-def process_input(git_url, personal_access_token, git_provider):
+def process_input(git_url, git_provider):
     global generated_file
     generated_file = None
 
     if not git_url.startswith(f"https://{git_provider.lower()}.com/"):
-        return f"Error: Invalid {git_provider} URL. Please use the format: https://{git_provider.lower()}.com/username/repository.git"
-    if not personal_access_token.strip():
-        return "Error: Personal Access Token is empty. Please provide a valid token."
+        return f"Error: Invalid {git_provider} URL. Please use the format: https://{git_provider.lower()}.com/username/repository.git", None
     
-    file_content = fetch_git_files(git_url, personal_access_token, git_provider)
+    file_content = fetch_git_files(git_url, git_provider)
     if file_content.startswith("Error:"):
-        return file_content
+        return file_content, None
     
     try:
         # Process the file content with Gemini
         analysis = process_with_gemini(file_content, GEMINI_API_KEY)
         generated_file = analysis.encode()
-        return "Analysis complete. Click the download button to get the results."
+        return "Analysis complete. You can now preview the results and download them.", analysis
     except Exception as e:
-        return f"Error processing the files: {str(e)}"
+        return f"Error processing the files: {str(e)}", None
 
 app.layout = dbc.Container([
-    html.H1("Open Source License Extractor", className="my-4"),
-    html.P("Provide a Git repository URL to analyze open-source licenses from dependency files.", className="mb-4"),
-    dbc.Card([
-        dbc.CardBody([
-            dbc.Row([
-                dbc.Col([
+    dbc.Row([
+        dbc.Col(html.H1("Open Source License Extractor", className="my-4"), width=12)
+    ]),
+    dbc.Row([
+        dbc.Col([
+            dbc.Card([
+                dbc.CardBody([
+                    html.P("Provide a Git repository URL to analyze open-source licenses from dependency files.", className="mb-4"),
                     dcc.Dropdown(
                         id='git-provider',
                         options=[
@@ -159,11 +161,7 @@ app.layout = dbc.Container([
                         className="mb-3"
                     ),
                     dbc.Input(id="git-url", placeholder="Enter Git Repository URL", type="text", className="mb-3"),
-                    dbc.Input(id="personal-access-token", placeholder="Enter Git Personal Access Token", type="password", className="mb-3"),
                     dbc.Button("Analyze", id="analyze-button", color="primary", className="mb-3"),
-                    dbc.Button("Download Results", id="download-button", color="secondary", className="mb-3 ml-2", disabled=True),
-                    dcc.Download(id="download-analysis"),
-                    html.Div(id="output", className="mt-3"),
                     dcc.Loading(
                         id="loading",
                         type="dot",
@@ -171,44 +169,71 @@ app.layout = dbc.Container([
                     )
                 ])
             ])
-        ])
+        ], md=6),
+        dbc.Col([
+            dbc.Card([
+                dbc.CardBody([
+                    html.H4("Analysis Results", className="mb-3"),
+                    html.Div(id="output", className="mb-3"),
+                    dbc.Button("Download as Word", id="download-word-button", color="secondary", className="mb-3 me-2", disabled=True),
+                    dbc.Button("Download as Markdown", id="download-markdown-button", color="secondary", className="mb-3", disabled=True),
+                    dcc.Download(id="download-word"),
+                    dcc.Download(id="download-markdown"),
+                ])
+            ])
+        ], md=6)
     ])
 ], fluid=True)
 
 @app.callback(
     [Output("output", "children"),
-     Output("download-button", "disabled"),
+     Output("download-word-button", "disabled"),
+     Output("download-markdown-button", "disabled"),
      Output("loading-output", "children")],
     [Input("analyze-button", "n_clicks")],
     [State("git-url", "value"),
-     State("personal-access-token", "value"),
      State("git-provider", "value")],
     prevent_initial_call=True
 )
-def update_output(n_clicks, git_url, personal_access_token, git_provider):
+def update_output(n_clicks, git_url, git_provider):
     if n_clicks is None:
         raise PreventUpdate
     
     def process():
         global generated_file
-        result = process_input(git_url, personal_access_token, git_provider)
-        return result, generated_file is not None, ""
+        result, analysis = process_input(git_url, git_provider)
+        return analysis if analysis else result, generated_file is None, generated_file is None, ""
 
     return process()
 
 @app.callback(
-    Output("download-analysis", "data"),
-    Input("download-button", "n_clicks"),
+    Output("download-word", "data"),
+    Input("download-word-button", "n_clicks"),
     prevent_initial_call=True
 )
-def download_analysis(n_clicks):
-    if n_clicks is None:
+def download_word(n_clicks):
+    if n_clicks is None or generated_file is None:
         raise PreventUpdate
+
+    doc = Document()
+    doc.add_paragraph(generated_file.decode())
+    
+    buffer = BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
     
-    if generated_file is None:
-        return dash.no_update
+    return dcc.send_bytes(buffer.getvalue(), "license_analysis.docx")
+
+@app.callback(
+    Output("download-markdown", "data"),
+    Input("download-markdown-button", "n_clicks"),
+    prevent_initial_call=True
+)
+def download_markdown(n_clicks):
+    if n_clicks is None or generated_file is None:
+        raise PreventUpdate
 
-    return dcc.send_bytes(generated_file, "license_analysis.txt")
+    return dcc.send_string(generated_file.decode(), "license_analysis.md")
 
 if __name__ == '__main__':
     print("Starting the Dash application...")