MMAPI-2

Running

App Files Files Community

DeFactOfficial commited on Oct 25

Commit

a6f672d

•

1 Parent(s): 3df0e86

fix paths and ports for HF setup

Browse files

Files changed (5) hide show

app.py +0 -52
log.txt +12 -0
mmapi/api.js +4 -4
mmapi/image.js +3 -4
mmapi/public/index.html +114 -0

app.py CHANGED Viewed

@@ -42,55 +42,3 @@ else:
     # Log the process spawn success
     with open(log_file_path, 'a') as log_file:
         log_file.write(f"[{datetime.now()}] API service started and running in the background.\n")
-# Gradio app to display the HTML documentation
-documentation_html = """
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>API Documentation</title>
-    <style>
-        body {
-            font-family: Arial, sans-serif;
-            line-height: 1.6;
-            max-width: 800px;
-            margin: 20px auto;
-            padding: 0 20px;
-            color: #333;
-        }
-        h1 {
-            color: #0056b3;
-        }
-        code {
-            background: #f4f4f4;
-            padding: 2px 6px;
-            border-radius: 4px;
-        }
-        pre {
-            background: #f4f4f4;
-            padding: 10px;
-            border-radius: 4px;
-            overflow-x: auto;
-        }
-    </style>
-</head>
-<body>
-    <h1>API Documentation</h1>
-    <p>This page documents the available APIs for the Node.js service, detailing endpoints, parameters, and expected responses.</p>
-    <!-- Content omitted for brevity -->
-</body>
-</html>
-"""
-# Create a Gradio interface that displays the HTML documentation
-def show_documentation():
-    return documentation_html
-doc_app = gr.Blocks()
-with doc_app:
-    gr.HTML(show_documentation())
-# Launch the Gradio app
-doc_app.launch(server_port=7861, server_name="0.0.0.0")

     # Log the process spawn success
     with open(log_file_path, 'a') as log_file:
         log_file.write(f"[{datetime.now()}] API service started and running in the background.\n")

log.txt CHANGED Viewed

@@ -24,3 +24,15 @@ Emitted 'error' event on Server instance at:
 }
 Node.js v20.18.0

 }
 Node.js v20.18.0
+[2024-10-25 05:17:24.382930] Script started.
+[2024-10-25 05:17:24.383335] API service already running on port 7860, exiting script.
+[2024-10-25 05:18:50.285983] Script started.
+[2024-10-25 05:18:50.286348] API service already running on port 7860, exiting script.
+[2024-10-25 05:19:34.842210] Script started.
+[2024-10-25 05:19:34.842602] API service already running on port 7860, exiting script.
+[2024-10-25 05:24:34.969499] Script started.
+[2024-10-25 05:24:34.969881] API service already running on port 7860, exiting script.

mmapi/api.js CHANGED Viewed

@@ -15,8 +15,8 @@ app.use(cors()); // Enable CORS for all routes
 require('dotenv').config()
-const MEDIA_FOLDER = process.env.MMAPI_MEDIA_FOLDER || 'media';
-const BASE_URL = process.env.MMAPI_BASE_URL || 'http://localhost:8000';
 const OPENAI_API_KEY = process.env.OPENAI_API_KEY
 // Ensure the MEDIA_FOLDER directory exists
@@ -274,8 +274,8 @@ app.post("/generate/image", async(req, res)=> {
-// Serve static files from the MEDIA_FOLDER
-app.use('/media', express.static(MEDIA_FOLDER));
 const port = 7860;
 app.listen(port, () => {

 require('dotenv').config()
+const MEDIA_FOLDER = `${process.env.STATIC_SITE_ROOT}/media`
+const BASE_URL = process.env.MMAPI_BASE_URL || 'http://localhost:7860';
 const OPENAI_API_KEY = process.env.OPENAI_API_KEY
 // Ensure the MEDIA_FOLDER directory exists
+// Client webpages and storage for generated content
+app.use('/', express.static(process.env.STATIC_SITE_ROOT));
 const port = 7860;
 app.listen(port, () => {

mmapi/image.js CHANGED Viewed

@@ -1,4 +1,3 @@
-const express = require('express');
 const axios = require('axios');
 const crypto = require('crypto');
 const fs = require('fs');
@@ -6,12 +5,12 @@ const path = require('path');
 const { createCanvas, loadImage } = require('canvas');
 const { HfInference } = require("@huggingface/inference");
-// Initialize the Express app
-const app = express();
 require("dotenv").config()
 // Cache directory to store generated images
-const CACHE_DIR = process.env.IMAGE_FOLDER || '/home/sam/disseminate/media/image_cache';
 if (!fs.existsSync(CACHE_DIR)) {
   fs.mkdirSync(CACHE_DIR);
 }

 const axios = require('axios');
 const crypto = require('crypto');
 const fs = require('fs');
 const { createCanvas, loadImage } = require('canvas');
 const { HfInference } = require("@huggingface/inference");
 require("dotenv").config()
 // Cache directory to store generated images
+const CACHE_DIR = `${process.env.STATIC_SITE_ROOT}/image_out`
 if (!fs.existsSync(CACHE_DIR)) {
   fs.mkdirSync(CACHE_DIR);
 }

mmapi/public/index.html ADDED Viewed

	@@ -0,0 +1,114 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>API Documentation</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            line-height: 1.6;
+            max-width: 800px;
+            margin: 20px auto;
+            padding: 0 20px;
+            color: #333;
+        }
+        h1 {
+            color: #0056b3;
+        }
+        code {
+            background: #f4f4f4;
+            padding: 2px 6px;
+            border-radius: 4px;
+        }
+        pre {
+            background: #f4f4f4;
+            padding: 10px;
+            border-radius: 4px;
+            overflow-x: auto;
+        }
+    </style>
+</head>
+<body>
+    <h1>API Documentation</h1>
+    <p>This page documents the available APIs for the Node.js service, detailing endpoints, parameters, and expected responses.</p>
+    <h2>1. GET /list-models</h2>
+    <p>Retrieves a list of available models.</p>
+    <strong>URL:</strong>
+    <pre>GET /list-models</pre>
+    <strong>Response:</strong>
+    <pre>["Model 1", "Model 2", "Model 3"]</pre>
+    <h2>2. GET /generate/speech</h2>
+    <p>Generates speech audio based on the provided script.</p>
+    <strong>URL:</strong>
+    <pre>GET /generate/speech?api_key=<code>your_api_key</code>&payload=<code>script_payload</code></pre>
+    <strong>Parameters:</strong>
+    <ul>
+        <li><strong>api_key</strong> (required): Your API key for authentication.</li>
+        <li><strong>payload</strong> (required): The script payload containing speaker names and speech content.</li>
+    </ul>
+    <strong>Response:</strong>
+    <pre>Audio file in MPEG format (served as a response file).</pre>
+    <h2>3. POST /generate/speech/stream</h2>
+    <p>Generates speech audio and streams the response via Server-Sent Events (SSE).</p>
+    <strong>URL:</strong>
+    <pre>POST /generate/speech/stream?api_key=<code>your_api_key</code></pre>
+    <strong>Headers:</strong>
+    <ul>
+        <li><strong>Content-Type</strong>: application/json</li>
+    </ul>
+    <strong>Body:</strong>
+    <pre>{
+    "payload": "script_payload"
+}</pre>
+    <strong>Response:</strong>
+    <ul>
+        <li><code>audio_segment</code>: URL to generated audio segment.</li>
+        <li><code>audio_complete</code>: URL to the final combined audio file.</li>
+    </ul>
+    <h2>4. GET /generate/image</h2>
+    <p>Generates an image based on the provided prompt.</p>
+    <strong>URL:</strong>
+    <pre>GET /generate/image?prompt=<code>image_prompt</code>&width=<code>image_width</code>&height=<code>image_height</code>&response_format=<code>image</code> (default)</pre>
+    <strong>Parameters:</strong>
+    <ul>
+        <li><strong>prompt</strong> (required): Description of the image to generate.</li>
+        <li><strong>width</strong> (optional): Width of the generated image (default is 1024).</li>
+        <li><strong>height</strong> (optional): Height of the generated image (default is 1024).</li>
+        <li><strong>response_format</strong> (optional): Format of the response, can be "image" or "url".</li>
+    </ul>
+    <strong>Response:</strong>
+    <pre>Generated image or URL to the image.</pre>
+    <h2>5. POST /generate/image</h2>
+    <p>Generates an image based on a prompt passed in the body of the request.</p>
+    <strong>URL:</strong>
+    <pre>POST /generate/image?response_format=<code>url</code> (default)</pre>
+    <strong>Headers:</strong>
+    <ul>
+        <li><strong>Content-Type</strong>: application/json</li>
+    </ul>
+    <strong>Body:</strong>
+    <pre>{
+    "prompt": "A large hamster"
+}</pre>
+    <strong>Response:</strong>
+    <pre>{ "imageUrl": "https://example.com/generated_image.jpg" }</pre>
+    <h2>6. Serving Static Media Files</h2>
+    <p>All generated media (audio and images) can be accessed via:</p>
+    <strong>URL:</strong>
+    <pre>GET /media/<code>filename</code></pre>
+    <h2>Notes</h2>
+    <ul>
+        <li>Ensure that your API key is provided when required.</li>
+        <li>Audio and image generation requires a valid script payload or image prompt respectively.</li>
+        <li>The system uses OpenAI for TTS, and appropriate API keys need to be set in the environment variables.</li>
+    </ul>
+</body>
+</html>